diff --git a/tests/test_global_observation.py b/tests/test_global_observation.py
index fe3e9ec1c2d4a98e386e108030de1cd725c213c7..9035062bad47dbdf432c6cd980c9d2361d42de6e 100644
--- a/tests/test_global_observation.py
+++ b/tests/test_global_observation.py
@@ -23,15 +23,14 @@ def test_get_global_observation():
 
     env = RailEnv(width=50,
                   height=50,
-                  rail_generator=sparse_rail_generator(max_num_cities=25,
+                  rail_generator=sparse_rail_generator(max_num_cities=6,
                                                        max_rails_between_cities=4,
                                                        seed=15,
-                                                       grid_mode=True
+                                                       grid_mode=False
                                                        ),
                   schedule_generator=sparse_schedule_generator(speed_ration_map),
                   number_of_agents=number_of_agents, stochastic_data=stochastic_data,  # Malfunction data generator
                   obs_builder_object=GlobalObsForRailEnv())
-
     obs, all_rewards, done, _ = env.step({0: 0})
 
     for i in range(len(env.agents)):