diff --git a/run.py b/run.py
index b7c3b9b781681fafbfd648e13623713c346ea77d..ed3afa0c9ddd593d4d360be1e631efcaa7febe40 100644
--- a/run.py
+++ b/run.py
@@ -26,8 +26,8 @@ from reinforcement_learning.dddqn_policy import DDDQNPolicy
 VERBOSE = True
 
 # Checkpoint to use (remember to push it!)
-checkpoint = "./checkpoints/201112143850-5400.pth" # 21.220418678677177 DEPTH=2 AGENTS=10
-# checkpoint = "./checkpoints/201113211844-6100.pth" # 19.690047767961005 DEPTH=2 AGENTS=20
+# checkpoint = "./checkpoints/201112143850-5400.pth" # 21.220418678677177 DEPTH=2 AGENTS=10
+checkpoint = "./checkpoints/201113211844-6100.pth" # 19.690047767961005 DEPTH=2 AGENTS=20
 
 
 # Use last action cache