diff --git a/07_ensemble_learning_and_random_forests.ipynb b/07_ensemble_learning_and_random_forests.ipynb
index 0aabfed..8854af5 100644
--- a/07_ensemble_learning_and_random_forests.ipynb
+++ b/07_ensemble_learning_and_random_forests.ipynb
@@ -327,11 +327,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from math import ceil, sqrt\n",
-    "\n",
     "bag_clf = BaggingClassifier(\n",
-    "    DecisionTreeClassifier(max_leaf_nodes=16),\n",
-    "    n_estimators=500, max_features=ceil(sqrt(X_train.shape[1])), random_state=42)"
+    "    DecisionTreeClassifier(max_features=\"sqrt\", max_leaf_nodes=16),\n",
+    "    n_estimators=500, random_state=42)"
    ]
   },
   {