diff --git a/test_submission.py b/test_submission.py
index 4ca3cd36bbb0bb99c445be7083e50ca638e5d28b..1cb3ff7aa268b1baa888b1d78678443ffc9f5692 100644
--- a/test_submission.py
+++ b/test_submission.py
@@ -7,7 +7,6 @@
 
 import numpy as np
 
-from agents.batched_agent import BatchedAgent
 from submission_config import SubmissionConfig, TestEvaluationConfig
 
 from rollout import run_batched_rollout