diff --git a/submission_config.py b/submission_config.py
index ba35352570cc0fbb94916ade55ce550422648dc1..2eb1e585fcbec5ec479747784f42d3567bceb246 100644
--- a/submission_config.py
+++ b/submission_config.py
@@ -14,8 +14,8 @@ from envs.wrappers import addtimelimitwrapper_fn
 
 class SubmissionConfig:
     ## Add your own agent class
-    AGENT = CustomAgent
-    # AGENT = TorchBeastAgent
+    # AGENT = CustomAgent
+    AGENT = TorchBeastAgent
 
 
     ## Change the NUM_ENVIRONMENTS as you need