diff --git a/base/src/com/google/idea/blaze/base/buildmodifier/BuildFileFormatter.java b/base/src/com/google/idea/blaze/base/buildmodifier/BuildFileFormatter.java
index d680f603ffc..8803597cbd8 100644
--- a/base/src/com/google/idea/blaze/base/buildmodifier/BuildFileFormatter.java
+++ b/base/src/com/google/idea/blaze/base/buildmodifier/BuildFileFormatter.java
@@ -93,7 +93,13 @@ static ImmutableList<String> getCommandLineArgs(String binary, BuildFile buildFi
   }
 
   private static String fileTypeArg(BlazeFileType fileType) {
-    return fileType == BlazeFileType.SkylarkExtension ? "--type=bzl" : "--type=build";
+    return "--type="
+        + switch (fileType) {
+          case SkylarkExtension -> "bzl";
+          case BuildPackage -> "build";
+          case Workspace -> "workspace";
+          case MODULE -> "module";
+        };
   }
 
   private static Iterable<String> pathArg(@Nullable BuildFile buildFile) {