diff --git a/userbenchmark/triton/ci.py b/userbenchmark/triton/ci.py
index aea648317c..7eb6be4abe 100644
--- a/userbenchmark/triton/ci.py
+++ b/userbenchmark/triton/ci.py
@@ -25,19 +25,19 @@
         "fp8_gemm",
         "--llama",
         "--only",
-        "torch_fp8_gemmm, triton_fp8_gemm",
+        "torch_fp8_gemm, triton_fp8_gemm",
     ],
     "fp8_gemm_blockwise": [
         "--op",
         "fp8_gemm_blockwise",
-        "--extended-shapes",
+        "--llama",
         "--only",
         "_cutlass, _triton",
     ],
     "fp8_gemm_rowwise": [
         "--op",
         "fp8_gemm_rowwise",
-        "--extended-shapes",
+        "--llama",
         "--only",
         "_cutlass,_triton,_cublass",
     ],