diff --git a/csrc/prepare_inputs/advance_step.cu b/csrc/prepare_inputs/advance_step.cu
index a9d08ca0..1f3f4710 100644
--- a/csrc/prepare_inputs/advance_step.cu
+++ b/csrc/prepare_inputs/advance_step.cu
@@ -211,7 +211,7 @@ void advance_step_flashinfer(
     printf("  num_seqs = %d\n", num_seqs);
     printf("  num_queries = %d\n", num_queries);
     printf("  block_size = %d\n", block_size);
-    printf("  block_tables.stride(0) = %d\n", block_tables.stride(0));
+    printf("  block_tables.stride(0) = %zu\n", block_tables.stride(0));
   }
   // Verify all tensors
   verify_tensor("input_tokens", input_tokens, num_seqs, -1, at::kLong);
@@ -303,4 +303,4 @@ void advance_step_flashinfer(
       num_seqs, num_queries, block_size, input_tokens, sampled_token_ids,
       input_positions, seq_lens, slot_mapping, block_tables, paged_kv_indices,
       paged_kv_indptr, paged_kv_last_page_len, block_table_bound);
-}
\ No newline at end of file
+}