diff --git a/examples/simple_example_3.py b/examples/simple_example_3.py
index 1661ef65a9a33f3b44a098caaf83317919722398..e015b3c88cf05a8d047f15dfaf88e8a2fd9ce789 100644
--- a/examples/simple_example_3.py
+++ b/examples/simple_example_3.py
@@ -2,7 +2,7 @@ import random
 
 import numpy as np
 
-from flatland.envs.generators import random_rail_generator
+from flatland.envs.generators import random_rail_generator, complex_rail_generator
 from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.rail_env import RailEnv
 from flatland.utils.rendertools import RenderTool