diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index 74032a9b766285715bc87037b46a9cb3332d0598..71a185c765bcab831e7b104124a164bcf2398b14 100644
--- a/examples/flatland_2_0_example.py
+++ b/examples/flatland_2_0_example.py
@@ -1,5 +1,5 @@
 import numpy as np
-from flatland.envs.generators import sparse_rail_generator
+from flatland.envs.rail_generators import sparse_rail_generator
 
 from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.predictions import ShortestPathPredictorForRailEnv