diff --git a/test/test_persistent_kernels.expected b/test/test_persistent_kernels.expected
index efb27a4dd..8be817280 100644
--- a/test/test_persistent_kernels.expected
+++ b/test/test_persistent_kernels.expected
@@ -2092,7 +2092,7 @@ def test_kernel(x: torch.Tensor, *, _launcher=_default_launcher):
     _BLOCK_SIZE_1 = 16
     # src[test_persistent_kernels.py:N]: for tile in hl.tile(x.size(), block_size=[32, 16]):
     # src[test_persistent_kernels.py:N]:     result[tile] = x[tile] + 1
-    _launcher(_helion_test_kernel, (_NUM_SM,), x, result, _NUM_SM, _BLOCK_SIZE_0, _BLOCK_SIZE_1, num_warps=4, num_stages=2)
+    _launcher(_helion_test_kernel, (_NUM_SM,), x, result, _NUM_SM, _BLOCK_SIZE_0, _BLOCK_SIZE_1, num_warps=4, num_stages=1)
     # src[test_persistent_kernels.py:N]: return result
     return result