diff --git a/examples/demo.py b/examples/demo.py
index e9f4e294320fbf94e6555a15ae2811b54cdbfc0f..933a9b1f5e00906f46e3800b2014e3d4aba60147 100644
--- a/examples/demo.py
+++ b/examples/demo.py
@@ -13,6 +13,8 @@ from flatland.utils.rendertools import RenderTool
 random.seed(1)
 np.random.seed(1)
 
+__file_dirname__ = os.path.dirname(os.path.realpath(__file__))
+
 
 class Scenario_Generator:
     @staticmethod
@@ -59,7 +61,7 @@ class Scenario_Generator:
         env = RailEnv(width=20,
                       height=20,
                       rail_generator=rail_from_list_of_saved_GridTransitionMap_generator(
-                          [filename]),
+                          [filename,
                       number_of_agents=number_of_agents)
         """
         if os.path.exists(filename):
@@ -133,37 +135,45 @@ if False:
     demo_001.run_demo()
     demo_001 = None
 
-    demo_000 = Demo(Scenario_Generator.load_scenario('./env-data/railway/example_network_000.pkl'))
+    demo_000 = Demo(Scenario_Generator.load_scenario(
+        os.path.join(__file_dirname__, '..', 'env-data', 'railway', 'example_network_000.pkl')))
     demo_000.run_demo()
     demo_000 = None
 
-    demo_001 = Demo(Scenario_Generator.load_scenario('./env-data/railway/example_network_001.pkl'))
+    demo_001 = Demo(Scenario_Generator.load_scenario(
+        os.path.join(__file_dirname__, '..', 'env-data', 'railway', 'example_network_001.pkl')))
     demo_001.run_demo()
     demo_001 = None
 
-    demo_002 = Demo(Scenario_Generator.load_scenario('./env-data/railway/example_network_002.pkl'))
+    demo_002 = Demo(Scenario_Generator.load_scenario(
+        os.path.join(__file_dirname__, '..', 'env-data', 'railway', 'example_network_002.pkl')))
     demo_002.run_demo()
     demo_002 = None
 
-    demo_flatland_000 = Demo(Scenario_Generator.load_scenario('./env-data/railway/example_flatland_000.pkl'))
+    demo_flatland_000 = Demo(
+        Scenario_Generator.load_scenario(
+            os.path.join(__file_dirname__, '..', 'env-data', 'railway', 'example_flatland_000.pkl')))
     demo_flatland_000.renderer.resize()
     demo_flatland_000.run_demo(60)
     demo_flatland_000 = None
 
-    demo_flatland_000 = Demo(Scenario_Generator.load_scenario('./env-data/railway/example_network_003.pkl'))
+    demo_flatland_000 = Demo(
+        Scenario_Generator.load_scenario(
+            os.path.join(__file_dirname__, '..', 'env-data', 'railway', 'example_network_003.pkl')))
     demo_flatland_000.renderer.resize()
     demo_flatland_000.set_max_framerate(5)
     demo_flatland_000.run_demo(30)
     demo_flatland_000 = None
 
-    demo_flatland_000 = Demo(Scenario_Generator.load_scenario('./env-data/railway/example_flatland_001.pkl'))
+    demo_flatland_000 = Demo(
+        Scenario_Generator.load_scenario(
+            os.path.join(__file_dirname__, '..', 'env-data', 'railway', 'example_flatland_001.pkl')))
     demo_flatland_000.renderer.resize()
-    demo_flatland_000.set_record_frames('./rendering/frame_{:04d}.bmp')
+    demo_flatland_000.set_record_frames(os.path.join(__file_dirname__, '..', 'rendering', 'frame_{:04d}.bmp'))
     demo_flatland_000.run_demo(60)
     demo_flatland_000 = None
 
-if True:
-    demo_001 = Demo(Scenario_Generator.load_scenario('./env-data/railway/complex_scene.pkl'))
-    demo_001.set_record_frames('./rendering/frame_{:04d}.bmp')
-    demo_001.run_demo(360)
-    demo_001 = None
+demo_001 = Demo(Scenario_Generator.load_scenario('./env-data/railway/complex_scene.pkl'))
+demo_001.set_record_frames('./rendering/frame_{:04d}.bmp')
+demo_001.run_demo(360)
+demo_001 = None