diff --git a/tests/test_lint.py b/tests/test_lint.py
index 5aef217..613d233 100644
--- a/tests/test_lint.py
+++ b/tests/test_lint.py
@@ -1,3 +1,3 @@
 def test_linting():
     import subprocess
-    subprocess.check_output(['flake8', 'tensorboardX'])
+    subprocess.check_output(['ruff', 'check', 'tensorboardX'])