diff --git a/machina/envs/gym_env.py b/machina/envs/gym_env.py
index 22aa8161..35e66b6c 100644
--- a/machina/envs/gym_env.py
+++ b/machina/envs/gym_env.py
@@ -33,6 +33,7 @@
 """
 
 import gym
+import gym.wrappers
 from machina import logger