diff --git a/scripts/quantize_nvfp4.py b/scripts/quantize_nvfp4.py
index c1462ac..f45d11d 100644
--- a/scripts/quantize_nvfp4.py
+++ b/scripts/quantize_nvfp4.py
@@ -214,9 +214,9 @@ def run_calibration(model_path, export_dir, calib_save_path, amax_snapshot_path,
         "hf_ptq.py",
         "--model", model_path,
         "--quant", QUANT,
-        "--calib", str(calib_size),
+        "--calib_size", str(calib_size),
         "--calib_seq", str(calib_seq),
-        "--kv_cache_quant", KV_CACHE_QUANT,
+        "--kv_cache_qformat", KV_CACHE_QUANT,
         "--tp", str(TP),
         "--export_path", export_dir,
         "--trust_remote_code",