diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml
index 8a96614d6a08f..d80ae054d6038 100644
--- a/.buildkite/test-pipeline.yaml
+++ b/.buildkite/test-pipeline.yaml
@@ -431,8 +431,9 @@ steps:
   source_file_dependencies:
   - csrc/
   - tests/kernels/core
+  - tests/kernels/test_top_k_per_row.py
   commands:
-    - pytest -v -s kernels/core
+    - pytest -v -s kernels/core kernels/test_top_k_per_row.py
 
 - label: Kernels Attention Test %N # 23min
   timeout_in_minutes: 35
diff --git a/tests/kernels/test_top_k_per_row.py b/tests/kernels/test_top_k_per_row.py
index f52cddc8c3707..ccef9d7123640 100644
--- a/tests/kernels/test_top_k_per_row.py
+++ b/tests/kernels/test_top_k_per_row.py
@@ -125,7 +125,6 @@ def test_top_k_per_row(
         indices,
         values,
         num_rows,
-        top_k,
         logits.stride(0),
         logits.stride(1),
     )