diff --git a/flatland/evaluators/client.py b/flatland/evaluators/client.py
index 2d4be5589c15f666b0341851276fd41b4336505c..1b743843ae64dec4a9df3bf2e58c437f8a58d1e3 100644
--- a/flatland/evaluators/client.py
+++ b/flatland/evaluators/client.py
@@ -11,7 +11,7 @@ import numpy as np
 import redis
 
 import flatland
-from envs.malfunction_generators import malfunction_from_file
+from flatland.envs.malfunction_generators import malfunction_from_file
 from flatland.envs.rail_env import RailEnv
 from flatland.envs.rail_generators import rail_from_file
 from flatland.envs.schedule_generators import schedule_from_file
diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py
index 5dd65935bea9edfeb88482b9627d3628479d7547..e2790c2bffd8a96ac25b07a8e698b6df1bdaa27c 100644
--- a/flatland/evaluators/service.py
+++ b/flatland/evaluators/service.py
@@ -16,7 +16,7 @@ import redis
 import timeout_decorator
 
 import flatland
-from envs.malfunction_generators import malfunction_from_file
+from flatland.envs.malfunction_generators import malfunction_from_file
 from flatland.core.env_observation_builder import DummyObservationBuilder
 from flatland.envs.agent_utils import RailAgentStatus
 from flatland.envs.rail_env import RailEnv