diff --git a/src/Envs/classic_control/Pendulum.jl b/src/Envs/classic_control/Pendulum.jl
index 79e83b1..47d78ec 100644
--- a/src/Envs/classic_control/Pendulum.jl
+++ b/src/Envs/classic_control/Pendulum.jl
@@ -32,7 +32,7 @@ function step!(env::PendulumEnv, u)
     g, m, l, dt = 10f0, 1f0, 1f0, env.dt
 
     v = clamp.(u, -env.max_torque, env.max_torque)
-    env.last_u = Tracker.data(v)[1] # for rendering
+    env.last_u = v[1] # for rendering
 
     costs = angle_normalize.(θ).^2 .+ 1f-1(θ̇ .^2) .+ 1f-3(u.^2)
 
diff --git a/src/Envs/env_wrapper.jl b/src/Envs/env_wrapper.jl
index 78c22e9..91fbdfc 100644
--- a/src/Envs/env_wrapper.jl
+++ b/src/Envs/env_wrapper.jl
@@ -50,13 +50,7 @@ Returns the observational state of the environment. The original state can
 be accessed by `\`env._env.state\``.
 """
 function state(env::EnvWrapper)
-	try
-		return _get_obs(env._env)
-	catch y
-		if isa(y, UndefVarError) || isa(y, MethodError)
-			return env._env.state
-		end
-	end
+	return _get_obs(env._env)
 end
 
 
@@ -65,4 +59,4 @@ function testmode!(env::EnvWrapper, val::Bool=true)
 end
 
 trainable(env::EnvWrapper) = env.train
-game_over(env::EnvWrapper) = env.done
\ No newline at end of file
+game_over(env::EnvWrapper) = env.done
diff --git a/src/Gym.jl b/src/Gym.jl
index ecb6a5c..d4c37c8 100644
--- a/src/Gym.jl
+++ b/src/Gym.jl
@@ -1,7 +1,6 @@
 module Gym
 
 using Flux
-using Flux.Tracker
 
 #Spaces
 include("Spaces/Space.jl")