diff --git a/setup.py b/setup.py
index 71242ee..e5e411f 100644
--- a/setup.py
+++ b/setup.py
@@ -23,6 +23,9 @@ setup(name='torchy_baselines',
             'sphinx',
             'sphinx-autobuild',
             'sphinx-rtd-theme'
+        ],
+        'render': [
+            'opencv-python'
         ]
       },
       description='Pytorch version of Stable Baselines, implementations of reinforcement learning algorithms.',
diff --git a/torchy_baselines/common/save_util.py b/torchy_baselines/common/save_util.py
index 30dd0b2..7b6a2be 100644
--- a/torchy_baselines/common/save_util.py
+++ b/torchy_baselines/common/save_util.py
@@ -131,4 +131,4 @@ def json_to_data(json_string, custom_objects=None):
         else:
             # Read as it is
             return_data[data_key] = data_item
-    return return_data
\ No newline at end of file
+    return return_data