diff --git a/learned_optimization/continuous_eval/run_eval_chief.py b/learned_optimization/continuous_eval/run_eval_chief.py
index 74ded07b..b04fb029 100644
--- a/learned_optimization/continuous_eval/run_eval_chief.py
+++ b/learned_optimization/continuous_eval/run_eval_chief.py
@@ -332,7 +332,7 @@ def get_single_task(cfgs):
       raise ValueError(f"Task name: {task_name} doesn't have a normalizer!")
 
     v = onp.mean(v, axis=0)
-    nv = norm_map[task_name](v)
+    nv = norm_map[task_name](v)  # pyrefly: ignore[bad-argument-type]
     # assume that the seeds have same inner step.
     max_steps = x[0, -1]
     nv = nv / max_steps