diff --git a/envlogger/testing/catch_env.py b/envlogger/testing/catch_env.py
index f3fc6f1..90892f1 100644
--- a/envlogger/testing/catch_env.py
+++ b/envlogger/testing/catch_env.py
@@ -58,7 +58,7 @@ def __init__(self, rows=10, columns=5, seed=1):
   def reset(self):
     """Returns the first `TimeStep` of a new episode."""
     self._reset_next_step = False
-    self._ball_x = self._rng.randint(self._columns)
+    self._ball_x = self._rng.randint(self._columns)  # pyrefly: ignore[bad-assignment]
     self._ball_y = 0
     self._paddle_x = self._columns // 2
     return dm_env.restart(self._observation())