diff --git a/notebooks/Simple_Rendering_Demo.ipynb b/notebooks/Simple_Rendering_Demo.ipynb
index 818b213042810a8896bcdc32cd24ead2d0510298..b291ba0917f594131345554a96c08461385a68c6 100644
--- a/notebooks/Simple_Rendering_Demo.ipynb
+++ b/notebooks/Simple_Rendering_Demo.ipynb
@@ -47,7 +47,8 @@
    "source": [
     "import flatland.core.env\n",
     "import flatland.utils.rendertools as rt\n",
-    "from flatland.envs.rail_env import RailEnv, random_rail_generator\n",
+    "from flatland.envs.rail_env import RailEnv\n",
+    "from flatland.envs.rail_generators import random_rail_generator\n",
     "from flatland.envs.observations import TreeObsForRailEnv\n",
     "from flatland.envs.predictions import ShortestPathPredictorForRailEnv"
    ]