diff --git a/DPI/utils.py b/DPI/utils.py
index 7867eeb..91e7b2e 100644
--- a/DPI/utils.py
+++ b/DPI/utils.py
@@ -193,6 +193,7 @@ def make_env(args):
             frame_skip=args.action_repeat,
             video_recording=args.save_video,
             video_recording_dir=args.work_dir,
+            version=args.version,
         )
     return env