diff --git a/examples/simple_example_3.py b/examples/simple_example_3.py
index e015b3c88cf05a8d047f15dfaf88e8a2fd9ce789..1661ef65a9a33f3b44a098caaf83317919722398 100644
--- a/examples/simple_example_3.py
+++ b/examples/simple_example_3.py
@@ -2,7 +2,7 @@ import random
 
 import numpy as np
 
-from flatland.envs.generators import random_rail_generator, complex_rail_generator
+from flatland.envs.generators import random_rail_generator
 from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.rail_env import RailEnv
 from flatland.utils.rendertools import RenderTool
diff --git a/tests/test_environments.py b/tests/test_environments.py
index aa24467dd1d548a2b68a408f300089ee8135c639..79160ce549155ea7dc8e905f015642d7b6ed5723 100644
--- a/tests/test_environments.py
+++ b/tests/test_environments.py
@@ -107,7 +107,6 @@ def test_rail_environment_single_agent():
             if prev_pos != pos:
                 valid_active_actions_done += 1
 
-
         # After 6 movements on this railway network, the train should be back
         # to its original height on the map.
         assert (initial_pos[0] == agent.position[0])
@@ -124,8 +123,10 @@ def test_rail_environment_single_agent():
                 _, _, dones, _ = rail_env.step({0: action})
                 done = dones['__all__']
 
+
 test_rail_environment_single_agent()
 
+
 def test_dead_end():
     transitions = Grid4Transitions([])