diff --git a/examples/custom_railmap_example.py b/examples/custom_railmap_example.py
index 04da66904fda1a58847a4acc510d7fc4e4e86887..6162b918734eb311752675e75e203d90e5558c1c 100644
--- a/examples/custom_railmap_example.py
+++ b/examples/custom_railmap_example.py
@@ -43,6 +43,7 @@ def custom_schedule_generator() -> ScheduleGenerator:
 env = RailEnv(width=6,
               height=4,
               rail_generator=custom_rail_generator(),
+              schedule_generator=custom_schedule_generator(),
               number_of_agents=1)
 
 env.reset()