diff --git a/test_lint.sh b/test_lint.sh
index d7de5ad9e..f5994c26d 100755
--- a/test_lint.sh
+++ b/test_lint.sh
@@ -1,5 +1,9 @@
 #!/usr/bin/env bash
 
 set -eou pipefail
+lint_command="${1:-check}"
+if [[ "$lint_command" == "ci" ]]; then
+    lint_command="check"
+fi
 
-./tools/bazel run //tools:buildifier@check
+./tools/bazel run //tools:buildifier@$lint_command