diff --git a/.github/workflows/ruff.yml b/.github/workflows/ruff.yml
index 8f8f5ee3..a3fc3b2f 100644
--- a/.github/workflows/ruff.yml
+++ b/.github/workflows/ruff.yml
@@ -31,4 +31,4 @@ jobs:
         ruff vllm tests
     - name: Spelling check with codespell
       run: |
-         codespell --toml pyproject.toml
\ No newline at end of file
+        codespell --toml pyproject.toml
\ No newline at end of file
diff --git a/tests/kernels/test_prefix_prefill.py b/tests/kernels/test_prefix_prefill.py
index 4d051593..2b35335a 100644
--- a/tests/kernels/test_prefix_prefill.py
+++ b/tests/kernels/test_prefix_prefill.py
@@ -36,8 +36,8 @@ def test_contexted_kv_attention(
         torch.cuda.manual_seed(0)
     torch.set_default_device(device)
 
-    # Need this, otherwise when we capture the graph the process for GPU 1 would run on both
-    # GPU0 and GPU1 and things would hang
+    # Need this, otherwise when we capture the graph the process for GPU 1 would
+    # run on both GPU0 and GPU1 and things would hang
     #
     # see also similar issue: https://github.com/Dao-AILab/flash-attention/issues/523
     torch.cuda.set_device(device)