diff --git a/DPI/dmc2gym/__init__.py b/DPI/dmc2gym/__init__.py
index ed7b897..92027ad 100644
--- a/DPI/dmc2gym/__init__.py
+++ b/DPI/dmc2gym/__init__.py
@@ -8,6 +8,7 @@ def make(
     resource_files,
     img_source,
     total_frames,
+    version,
     seed=1,
     visualize_reward=True,
     from_pixels=False,
@@ -20,7 +21,7 @@ def make(
     video_recording=False,
     video_recording_dir=None,
 ):
-    env_id = 'dmc_%s_%s_%s-v1' % (domain_name, task_name, seed)
+    env_id = 'dmc_%s_%s_%s-v1' % (domain_name, task_name, version)
 
     if from_pixels:
         assert not visualize_reward, 'cannot use visualize reward when learning from pixels'