diff --git a/tests/test_flatland_schedule_generators.py b/tests/test_flatland_schedule_from_file.py
similarity index 100%
rename from tests/test_flatland_schedule_generators.py
rename to tests/test_flatland_schedule_from_file.py
diff --git a/tests/test_utils.py b/tests/test_utils.py
index bb344962f84ab6b651e9cd688f7784804e8062c2..cae9b7fb814d63dcbd8c40678a6fb02e46d56a3b 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -3,15 +3,13 @@ from typing import List, Tuple, Optional
 
 import numpy as np
 from attr import attrs, attrib
-from flatland.envs.malfunction_generators import MalfunctionParameters, malfunction_from_params
-
-from flatland.envs.rail_generators import RailGenerator
-
-from flatland.envs.schedule_generators import ScheduleGenerator
 
 from flatland.core.grid.grid4 import Grid4TransitionsEnum
 from flatland.envs.agent_utils import EnvAgent, RailAgentStatus
+from flatland.envs.malfunction_generators import MalfunctionParameters, malfunction_from_params
 from flatland.envs.rail_env import RailEnvActions, RailEnv
+from flatland.envs.rail_generators import RailGenerator
+from flatland.envs.schedule_generators import ScheduleGenerator
 from flatland.utils.rendertools import RenderTool
 
 
@@ -139,8 +137,6 @@ def run_replay_config(env: RailEnv, test_configs: List[ReplayConfig], rendering:
 
 
 def create_and_save_env(file_name: str, schedule_generator: ScheduleGenerator, rail_generator: RailGenerator):
-
-
     stochastic_data = MalfunctionParameters(malfunction_rate=1000,  # Rate of malfunction occurence
                                             min_duration=15,  # Minimal duration of malfunction
                                             max_duration=50  # Max duration of malfunction