diff --git a/02_end_to_end_machine_learning_project.ipynb b/02_end_to_end_machine_learning_project.ipynb
index 6765e86..e65205d 100644
--- a/02_end_to_end_machine_learning_project.ipynb
+++ b/02_end_to_end_machine_learning_project.ipynb
@@ -2247,7 +2247,7 @@
    },
    "outputs": [],
    "source": [
-    "housing_num = housing.drop(\"ocean_proximity\", axis=1)"
+    "housing_num = housing.select_dtypes(include=[np.number])"
    ]
   },
   {
@@ -2646,7 +2646,7 @@
     }
    ],
    "source": [
-    "housing_cat = housing[\"ocean_proximity\"]\n",
+    "housing_cat = housing.select_dtypes(exclude=[np.number])\n",
     "housing_cat.head(10)"
    ]
   },