diff --git a/tests/ci/test_controller.py b/tests/ci/test_controller.py
index e36046d42..0b23dabff 100644
--- a/tests/ci/test_controller.py
+++ b/tests/ci/test_controller.py
@@ -335,9 +335,6 @@ class TestControllerIntegration:
 		# Create wait action for 1 second - fix to use a dictionary
 		wait_action = {'wait': {'seconds': 5}}  # Corrected format
 
-		class WaitActionModel(ActionModel):
-			wait: dict | None = None
-
 		# Record start time
 		start_time = time.time()