diff --git a/polecart/main/main.py b/polecart/main/main.py
index cd7ca59..8debc5a 100755
--- a/polecart/main/main.py
+++ b/polecart/main/main.py
@@ -74,7 +74,7 @@ if __name__ == '__main__':
 # CartPole-v1
 # Acrobot-v1
 agent = Agent(
-	env_name = "Acrobot-v1",
+	env_name = "CartPole-v1",
 	network = DQN,
 	BATCH_SIZE = 128,
 	TAU = 0.005,