diff --git a/test/registered/jit/benchmark/bench_fused_eh_norm.py b/test/registered/jit/benchmark/bench_fused_eh_norm.py
index 4069e69d7..9744c0750 100644
--- a/test/registered/jit/benchmark/bench_fused_eh_norm.py
+++ b/test/registered/jit/benchmark/bench_fused_eh_norm.py
@@ -6,7 +6,9 @@ from sglang.jit_kernel.benchmark import marker
 from sglang.jit_kernel.fused_eh_norm import fused_eh_norm
 from sglang.test.ci.ci_register import register_cuda_ci
 
-register_cuda_ci(est_time=6, suite="base-b-kernel-benchmark-1-gpu-large")
+register_cuda_ci(
+    est_time=6, stage="base-b-kernel-benchmark", runner_config="1-gpu-large"
+)
 
 EPS = 1e-6
 
diff --git a/test/registered/jit/test_fused_eh_norm.py b/test/registered/jit/test_fused_eh_norm.py
index 3f2d1b4af..967fe77fa 100644
--- a/test/registered/jit/test_fused_eh_norm.py
+++ b/test/registered/jit/test_fused_eh_norm.py
@@ -6,7 +6,7 @@ import torch
 from sglang.jit_kernel.fused_eh_norm import fused_eh_norm
 from sglang.test.ci.ci_register import register_cuda_ci
 
-register_cuda_ci(est_time=45, suite="base-b-kernel-unit-1-gpu-large")
+register_cuda_ci(est_time=45, stage="base-b-kernel-unit", runner_config="1-gpu-large")
 register_cuda_ci(est_time=120, suite="nightly-kernel-1-gpu", nightly=True)
 
 pytestmark = pytest.mark.skipif(