diff --git a/tests/test_flatland_envs_rail_env.py b/tests/test_flatland_envs_rail_env.py
index 664c1155b5a00de487ec7db249ce86abee82febd..07cf778925864e2f1f871bf4621d9c4a65bca220 100644
--- a/tests/test_flatland_envs_rail_env.py
+++ b/tests/test_flatland_envs_rail_env.py
@@ -40,8 +40,11 @@ def test_save_load():
     agent_2_pos = env.agents[1].position
     agent_2_dir = env.agents[1].direction
     agent_2_tar = env.agents[1].target
-    #env.save("test_save.dat")
+    
+    env.save("test_save_2.pkl")
     RailEnvPersister.save(env, "test_save.pkl")
+    
+
     #env.load("test_save.dat")
     env, env_dict = RailEnvPersister.load_new("test_save.pkl")
     assert (env.width == 10)