diff --git a/python/tests/test_benchmark.py b/python/tests/test_benchmark.py
index bfcdd3f310..e869335e87 100644
--- a/python/tests/test_benchmark.py
+++ b/python/tests/test_benchmark.py
@@ -95,7 +95,7 @@ def setup():
         return (dt,), dict(max_concurrent_tasks=max_tasks)
 
     def func(dt, max_concurrent_tasks):
-        return dt.optimize(
+        return dt.optimize.compact(
             max_concurrent_tasks=max_concurrent_tasks, target_size=1024 * 1024 * 1024
         )