diff --git a/tests/test_env_observation_builder.py b/tests/test_env_observation_builder.py
index 1692a98982a787ecbadf79feb7a35593f82522e4..13f8d8f9d7df1808bfb69261ee25bf06f9da2d45 100644
--- a/tests/test_env_observation_builder.py
+++ b/tests/test_env_observation_builder.py
@@ -3,7 +3,7 @@
 
 import numpy as np
 
-from flatland.core.env_observation_builder import GlobalObsForRailEnv
+from flatland.envs.observations import GlobalObsForRailEnv
 from flatland.core.transition_map import GridTransitionMap, Grid4Transitions
 from flatland.envs.rail_env import RailEnv
 from flatland.envs.generators import rail_from_GridTransitionMap_generator