diff --git a/qwen3_6_scripts/ex_engine/python/patch_vllm_hot_path.py b/qwen3_6_scripts/ex_engine/python/patch_vllm_hot_path.py
index 2eefc59a..1829846c 100644
--- a/qwen3_6_scripts/ex_engine/python/patch_vllm_hot_path.py
+++ b/qwen3_6_scripts/ex_engine/python/patch_vllm_hot_path.py
@@ -164,7 +164,8 @@ def apply(strict=True):
             import vllm._custom_ops as ops
 
             def patched_reshape_and_cache(key, value, key_cache, value_cache,
-                                          slot_mapping, kv_cache_dtype, kv_scale):
+                                          slot_mapping, kv_cache_dtype,
+                                          k_scale, v_scale):
                 xllm_ops.reshape_and_cache(key, value, key_cache, value_cache,
                                            slot_mapping)