diff --git a/python/mscclpp_benchmark/mscclpp_op.py b/python/mscclpp_benchmark/mscclpp_op.py
index 60ecf0a12..0a105808d 100644
--- a/python/mscclpp_benchmark/mscclpp_op.py
+++ b/python/mscclpp_benchmark/mscclpp_op.py
@@ -208,7 +208,7 @@ def __init__(
         self.set_params(nblocks, block_size)
 
     def __call__(self, stream):
-        self.kernel.launch_kernel(self.params, 24, 1024, 0, stream)
+        self.kernel.launch_kernel(self.params, self.nblocks, self.block_size, 0, stream)
         return self.memory
 
     def set_params(self, nblocks, block_size):