diff --git a/examples/DiversionMode.ipynb b/examples/DiversionMode.ipynb
index dde993c..4d112f3 100644
--- a/examples/DiversionMode.ipynb
+++ b/examples/DiversionMode.ipynb
@@ -13,7 +13,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -24,7 +24,8 @@
     "urllib3.disable_warnings()\n",
     "\n",
     "# Import Hydrogenerate\n",
-    "from HydroGenerate.hydropower_potential import *"
+    "from HydroGenerate.hydropower_potential import *\n",
+    "from HydroGenerate.summary_results import *"
    ]
   },
   {