diff --git a/tutorials/Tutorial 1 - Training and Evaluation of Logistic Regression on Encrypted Data.ipynb b/tutorials/Tutorial 1 - Training and Evaluation of Logistic Regression on Encrypted Data.ipynb
index 2743a20a..ba36d6c0 100644
--- a/tutorials/Tutorial 1 - Training and Evaluation of Logistic Regression on Encrypted Data.ipynb	
+++ b/tutorials/Tutorial 1 - Training and Evaluation of Logistic Regression on Encrypted Data.ipynb	
@@ -94,7 +94,7 @@
     "    data = grouped.apply(lambda x: x.sample(grouped.size().min(), random_state=73).reset_index(drop=True))\n",
     "    # extract labels\n",
     "    y = torch.tensor(data[\"TenYearCHD\"].values).float().unsqueeze(1)\n",
-    "    data = data.drop(\"TenYearCHD\", 'columns')\n",
+    "    data = data.drop(\"TenYearCHD\", axis=1)\n",
     "    # standardize data\n",
     "    data = (data - data.mean()) / data.std()\n",
     "    x = torch.tensor(data.values).float()\n",