diff --git a/flatland/envs/predictions.py b/flatland/envs/predictions.py
index ecc0f8f18d3a2d9cb730c9fcadf237e7ce133a6d..3338e68126ee9a198e0208e17b1986f1c9fde6c8 100644
--- a/flatland/envs/predictions.py
+++ b/flatland/envs/predictions.py
@@ -4,8 +4,8 @@ Collection of environment-specific PredictionBuilder.
 
 import numpy as np
 
-from envs.rail_env import RailEnvActions
 from flatland.core.env_prediction_builder import PredictionBuilder
+from flatland.envs.rail_env import RailEnvActions
 
 
 class DummyPredictorForRailEnv(PredictionBuilder):