diff --git a/examples/DiversionMode.ipynb b/examples/DiversionMode.ipynb
index dde993c..bc8e3b5 100644
--- a/examples/DiversionMode.ipynb
+++ b/examples/DiversionMode.ipynb
@@ -25,6 +25,7 @@
     "\n",
     "# Import Hydrogenerate\n",
     "from HydroGenerate.hydropower_potential import *"
+    "from HydroGenerate.summary_results import *"
    ]
   },
   {