diff --git a/README.md b/README.md
index d29d253..c062c9d 100644
--- a/README.md
+++ b/README.md
@@ -14,9 +14,9 @@ http://arxiv.org/abs/1911.12247
 
 * Python 3.6 or 3.7
 * PyTorch version 1.2
-* OpenAI Gym version: 0.12.0 `pip install gym==0.12.0`
+* OpenAI Gym version: 0.17.2 `pip install gym==0.17.2`
 * OpenAI Atari_py version: 0.1.4: `pip install atari-py==0.1.4`
-* Scikit-image version 0.15.0 `pip install scikit-image==0.15.0`
+* Scikit-image version 0.17.2 `pip install scikit-image==0.17.2`
 * Matplotlib version 3.0.2 `pip install matplotlib==3.0.2`
 
 ## Generate datasets
diff --git a/envs/block_pushing.py b/envs/block_pushing.py
index 94c5843..708957e 100644
--- a/envs/block_pushing.py
+++ b/envs/block_pushing.py
@@ -13,8 +13,7 @@
 from mpl_toolkits.mplot3d import Axes3D
 from PIL import Image
 
-
-import skimage
+import skimage.draw
 
 
 def square(r0, c0, width, im_size):
@@ -206,4 +205,4 @@ def step(self, action):
 
         state_obs = (self.get_state(), self.render())
 
-        return state_obs, reward, done, None
+        return state_obs, reward, done, {}