diff --git a/tests/e2e/test_engine_perception.py b/tests/e2e/test_engine_perception.py
index 42f8ec9..0e5d595 100644
--- a/tests/e2e/test_engine_perception.py
+++ b/tests/e2e/test_engine_perception.py
@@ -506,4 +506,4 @@ class TestSAELearning:
         assert result is True
         mock_store_screen.assert_called_once()
         args, kwargs = mock_store_screen.call_args
-        assert args[1] == "NORMAL"
+        assert args[2] == "NORMAL"