diff --git a/flashinfer/decode.py b/flashinfer/decode.py
index 467152af38..45bc2c58ad 100644
--- a/flashinfer/decode.py
+++ b/flashinfer/decode.py
@@ -1988,7 +1988,7 @@ def paged_run(
             q.contiguous(),  # NOTE(Siyuan): without contiguous, the result is incorrect
             paged_k_cache,
             paged_v_cache,
-            int_workspace_buffer,
+            float_workspace_buffer,
             block_tables,
             kv_lens_buffer,
             max_kv_len,