diff --git a/L3_DataPreparation.ipynb b/L3_DataPreparation.ipynb
index bad438c..de684f8 100644
--- a/L3_DataPreparation.ipynb
+++ b/L3_DataPreparation.ipynb
@@ -1247,7 +1247,7 @@
     "import syft as sy\n",
     "\n",
     "# select all of Country 0's data\n",
-    "country0_data = raw_data[0, :]\n",
+    "country0_data = raw_data[:, 0]\n",
     "\n",
     "# Specify it to be a np.int32 dtype\n",
     "country0_data = country0_data.astype(np.int32)\n",