diff --git a/iris/algorithms/multi_agent_ars_algorithm_test.py b/iris/algorithms/multi_agent_ars_algorithm_test.py
index 1cc662d..25c5b28 100644
--- a/iris/algorithms/multi_agent_ars_algorithm_test.py
+++ b/iris/algorithms/multi_agent_ars_algorithm_test.py
@@ -523,15 +523,15 @@ def test_split_checkpoint(self):
       if expected_states[i]['obs_norm_state'] is not None:
         self.assertAllClose(
             agent_state['obs_norm_state']['mean'],
-            expected_states[i]['obs_norm_state']['mean'],
+            expected_states[i]['obs_norm_state']['mean'],  # pyrefly: ignore[bad-index]
         )
         self.assertAllClose(
             agent_state['obs_norm_state']['std'],
-            expected_states[i]['obs_norm_state']['std'],
+            expected_states[i]['obs_norm_state']['std'],  # pyrefly: ignore[bad-index]
         )
         self.assertAllClose(
             agent_state['obs_norm_state']['n'],
-            expected_states[i]['obs_norm_state']['n'],
+            expected_states[i]['obs_norm_state']['n'],  # pyrefly: ignore[bad-index]
         )