diff --git a/02_end_to_end_machine_learning_project.ipynb b/02_end_to_end_machine_learning_project.ipynb
index 39d9a38..dc660a7 100644
--- a/02_end_to_end_machine_learning_project.ipynb
+++ b/02_end_to_end_machine_learning_project.ipynb
@@ -753,7 +753,7 @@
    "outputs": [],
    "source": [
     "housing_tr = pd.DataFrame(X, columns=housing_num.columns,\n",
-    "                          index = list(housing.index.values))"
+    "                          index=housing.index)"
    ]
   },
   {
@@ -780,7 +780,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "housing_tr = pd.DataFrame(X, columns=housing_num.columns)"
+    "housing_tr = pd.DataFrame(X, columns=housing_num.columns,\n",
+    "                          index=housing_num.index)"
    ]
   },
   {
@@ -932,7 +933,8 @@
    "source": [
     "housing_extra_attribs = pd.DataFrame(\n",
     "    housing_extra_attribs,\n",
-    "    columns=list(housing.columns)+[\"rooms_per_household\", \"population_per_household\"])\n",
+    "    columns=list(housing.columns)+[\"rooms_per_household\", \"population_per_household\"],\n",
+    "    index=housing.index)\n",
     "housing_extra_attribs.head()"
    ]
   },