diff --git a/examples/simple_example_3.py b/examples/simple_example_3.py
index b5283df627b89294b899a0dc3f4652d5d2375152..1978e27c8511154d317ef5c96c3a9bc6816168c5 100644
--- a/examples/simple_example_3.py
+++ b/examples/simple_example_3.py
@@ -3,7 +3,7 @@ import random
 from flatland.envs.generators import random_rail_generator
 from flatland.envs.rail_env import RailEnv
 from flatland.utils.rendertools import RenderTool
-from flatland.core.env_observation_builder import TreeObsForRailEnv
+from flatland.envs.observations import TreeObsForRailEnv
 import numpy as np
 
 random.seed(100)