diff --git a/02_end_to_end_machine_learning_project.ipynb b/02_end_to_end_machine_learning_project.ipynb
index b0f72e7..e1ee729 100644
--- a/02_end_to_end_machine_learning_project.ipynb
+++ b/02_end_to_end_machine_learning_project.ipynb
@@ -835,8 +835,8 @@
     "splitter = StratifiedShuffleSplit(n_splits=10, test_size=0.2, random_state=42)\n",
     "strat_splits = []\n",
     "for train_index, test_index in splitter.split(housing, housing[\"income_cat\"]):\n",
-    "    strat_train_set_n = housing.loc[train_index]\n",
-    "    strat_test_set_n = housing.loc[test_index]\n",
+    "    strat_train_set_n = housing.iloc[train_index]\n",
+    "    strat_test_set_n = housing.iloc[test_index]\n",
     "    strat_splits.append([strat_train_set_n, strat_test_set_n])"
    ]
   },