diff --git a/spada/runtime/runtime.py b/spada/runtime/runtime.py
index 9ef3adc6..394c8a66 100644
--- a/spada/runtime/runtime.py
+++ b/spada/runtime/runtime.py
@@ -389,7 +389,7 @@ def __call__(self, *args, **kwargs) -> Dict[str, np.ndarray]:
             print("Copy-back complete.", flush=True)
 
             if self.benchmark and not self.metadata.memcpy_mode:
-                cycle_counts = copy_back_benchmark_data(self.runtime, self.metadata)
+                cycle_counts = copy_back_benchmark_cycles(self.runtime, self.metadata)
                 np.save(self.output_dir / "perf_cycles.npy", cycle_counts)
                 print_cycle_counts("Cycle count", cycle_counts)