diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index ff21046bf22c59235a9d8c76656c9f368b1e799a..907b4a25edc0d44ff83d7a063795b55e020362b3 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -197,7 +197,7 @@ def rail_from_manual_specifications_generator(rail_spec):
     return generator
 
 
-def rail_from_data(filename):
+def rail_from_file(filename):
     """
     Utility to load pickle file
 
diff --git a/tests/test_file_load.py b/tests/test_file_load.py
index 2b929b174cca8666579e7a5c1eccce05a1a19176..57fa45cb29dab07b84f43c97a45043c9dfa39979 100644
--- a/tests/test_file_load.py
+++ b/tests/test_file_load.py
@@ -3,7 +3,7 @@
 
 import numpy as np
 
-from flatland.envs.generators import rail_from_GridTransitionMap_generator, rail_from_data
+from flatland.envs.generators import rail_from_GridTransitionMap_generator, rail_from_file
 from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.predictions import ShortestPathPredictorForRailEnv
 from flatland.envs.rail_env import RailEnv
@@ -26,7 +26,7 @@ def test_load_pkl():
 
     env = RailEnv(width=1,
                   height=1,
-                  rail_generator=rail_from_data(file_name),
+                  rail_generator=rail_from_file(file_name),
                   number_of_agents=1,
                   obs_builder_object=TreeObsForRailEnv(max_depth=2, predictor=ShortestPathPredictorForRailEnv()),
                   )