diff --git a/notebooks/simple_example1_env_from_tuple.ipynb b/notebooks/simple_example1_env_from_tuple.ipynb
index 3fd55bc8fafabdd57eb43e012c5d98b37c73d496..0fcfe26325e5778cbbfdf66591346972edb2406d 100644
--- a/notebooks/simple_example1_env_from_tuple.ipynb
+++ b/notebooks/simple_example1_env_from_tuple.ipynb
@@ -14,7 +14,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from flatland.envs.generators import rail_from_manual_specifications_generator\n",
+    "from flatland.envs.rail_generators import rail_from_manual_specifications_generator\n",
     "from flatland.envs.observations import TreeObsForRailEnv\n",
     "from flatland.envs.rail_env import RailEnv\n",
     "from flatland.utils.rendertools import RenderTool\n",
diff --git a/notebooks/simple_example2_generate_random_rail.ipynb b/notebooks/simple_example2_generate_random_rail.ipynb
index b9d4a96c02d4cb63392654025c6857c8b6764d1a..19b854ee15d8dd1e19361f58a552eba617b19b67 100644
--- a/notebooks/simple_example2_generate_random_rail.ipynb
+++ b/notebooks/simple_example2_generate_random_rail.ipynb
@@ -15,7 +15,7 @@
    "source": [
     "import random\n",
     "import numpy as np\n",
-    "from flatland.envs.generators import random_rail_generator\n",
+    "from flatland.envs.rail_generators import random_rail_generator\n",
     "from flatland.envs.observations import TreeObsForRailEnv\n",
     "from flatland.envs.rail_env import RailEnv\n",
     "from flatland.utils.rendertools import RenderTool\n",
diff --git a/notebooks/simple_example_3_manual_control.ipynb b/notebooks/simple_example_3_manual_control.ipynb
index 50f228055b320c15e0411c2b086254a1f4d4ceef..cb2b377765f375e8d77d8374fdcc3bb67ce06444 100644
--- a/notebooks/simple_example_3_manual_control.ipynb
+++ b/notebooks/simple_example_3_manual_control.ipynb
@@ -40,7 +40,7 @@
     "import random\n",
     "import numpy as np\n",
     "import time\n",
-    "from flatland.envs.generators import random_rail_generator\n",
+    "from flatland.envs.rail_generators import random_rail_generator\n",
     "from flatland.envs.observations import TreeObsForRailEnv\n",
     "from flatland.envs.rail_env import RailEnv\n",
     "from flatland.utils.rendertools import RenderTool"