diff --git a/pyrobolearn/envs/gym_wrapper.py b/pyrobolearn/envs/gym_wrapper.py
index dfc9303..634a64c 100644
--- a/pyrobolearn/envs/gym_wrapper.py
+++ b/pyrobolearn/envs/gym_wrapper.py
@@ -11,7 +11,7 @@ import functools
 import numpy as np
 import torch
 import gym
-import baselines
+# import baselines
 from gym import *
 import warnings
 warnings.simplefilter("ignore")