diff --git a/ProgrammingAssignment_2/model.ipynb b/ProgrammingAssignment_2/model.ipynb
index c37e9bd290d748385d0517d4a5f970e422c765af..8de6ead1eaca3e525e1c0195c208da1cc24ca042 100644
--- a/ProgrammingAssignment_2/model.ipynb
+++ b/ProgrammingAssignment_2/model.ipynb
@@ -104,7 +104,7 @@
     "            Remaining 1-(1/k) of the indices.\n",
     "            \n",
     "            e.g. fold_dict = {0: (train_0_indices, val_0_indices), \n",
-    "            1: (train_0_indices, val_0_indices), 2: (train_0_indices, val_0_indices)} for k = 3\n",
+    "            1: (train_1_indices, val_1_indices), 2: (train_2_indices, val_2_indices)} for k = 3\n",
     "    '''\n",
     "    \n",
     "    return fold_dict"