diff --git a/notebooks/parameter_tuning_grid_search.ipynb b/notebooks/parameter_tuning_grid_search.ipynb
index a7fc56994..79f2b49f3 100644
--- a/notebooks/parameter_tuning_grid_search.ipynb
+++ b/notebooks/parameter_tuning_grid_search.ipynb
@@ -216,8 +216,15 @@
     "explore for each parameter. Thus, adding new parameters with their associated\n",
     "values to be explored rapidly becomes computationally expensive. Because of\n",
     "that, here we only explore the combination learning-rate and the maximum\n",
-    "number of nodes for a total of 4 x 3 = 12 combinations.\n",
-    "\n",
+    "number of nodes for a total of 4 x 3 = 12 combinations."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "%%time\n",
     "from sklearn.model_selection import GridSearchCV\n",
     "\n",
diff --git a/python_scripts/parameter_tuning_grid_search.py b/python_scripts/parameter_tuning_grid_search.py
index d3cc4ecd4..30979909a 100644
--- a/python_scripts/parameter_tuning_grid_search.py
+++ b/python_scripts/parameter_tuning_grid_search.py
@@ -136,6 +136,7 @@
 # that, here we only explore the combination learning-rate and the maximum
 # number of nodes for a total of 4 x 3 = 12 combinations.
 
+# %%
 # %%time
 from sklearn.model_selection import GridSearchCV