diff --git a/ProgrammingAssignment_0/GettingFamiliar.ipynb b/ProgrammingAssignment_0/GettingFamiliar.ipynb
index 2cc5c53d51aea73b56d84adc985f8f7e2c70a4f1..3a9e674437409211753e4da94dfb91e1f05ca2b2 100644
--- a/ProgrammingAssignment_0/GettingFamiliar.ipynb
+++ b/ProgrammingAssignment_0/GettingFamiliar.ipynb
@@ -244,7 +244,7 @@
    "source": [
     "# TODO\n",
     "# Pass the correct size argument (number of examples in the whole dataset)\n",
-    "test_indices, val_indices = partition(size=..., t = 0.3, v = 0.1)\n",
+    "test_indices, val_indices, train_indices = partition(size=..., t = 0.3, v = 0.1)\n",
     "\n",
     "# Output the size of length of test and validation indices."
    ]