diff --git a/buildifier/buildifier.go b/buildifier/buildifier.go
index 57478c94d..5ca0b59a5 100644
--- a/buildifier/buildifier.go
+++ b/buildifier/buildifier.go
@@ -136,9 +136,7 @@ func main() {
 
 	diff = differ.Find()
 
-	// TODO(bazel-team): Handle "-" as stdin/stdout mode too.
-
-	if len(args) == 0 {
+	if len(args) == 0 || (len(args) == 1 && args[0] == "-") {
 		// Read from stdin, write to stdout.
 		data, err := ioutil.ReadAll(os.Stdin)
 		if err != nil {