diff --git a/tinygrad/features/search.py b/tinygrad/features/search.py
index 9d402cf824..58f1ce1ac6 100644
--- a/tinygrad/features/search.py
+++ b/tinygrad/features/search.py
@@ -45,7 +45,7 @@ def time_linearizer(lin:Linearizer, rawbufs:List[RawBuffer], allow_test_size=Tru
     # TODO: this is copied from prg.__call__
     global_size, local_size = prg.launch_dims(var_vals)
     if global_size is not None and local_size is None:
-      local_size = prg.optimize_local_size(global_size, rawbufs)
+      local_size = optimize_local_size(prg, global_size, rawbufs)
       global_size = [g//l if g%l == 0 else g/l for g,l in zip(global_size, local_size)]
     tms = []
     for _ in range(cnt):