diff --git a/utils/bazel/llvm-project-overlay/llvm/BUILD.bazel b/utils/bazel/llvm-project-overlay/llvm/BUILD.bazel
index dfed324a9cb49..79ca465e15875 100644
--- a/utils/bazel/llvm-project-overlay/llvm/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/llvm/BUILD.bazel
@@ -5919,6 +5919,7 @@ py_binary(
         "utils/mlgo-utils/mlgo/__init__.py",
         "utils/mlgo-utils/mlgo/corpus/extract_ir.py",
         "utils/mlgo-utils/mlgo/corpus/extract_ir_lib.py",
+        "utils/mlgo-utils/mlgo/corpus/flags.py",
     ],
     imports = ["utils/mlgo-utils"],
 )
@@ -5929,6 +5930,7 @@ py_binary(
         "utils/mlgo-utils/mlgo/__init__.py",
         "utils/mlgo-utils/mlgo/corpus/combine_training_corpus.py",
         "utils/mlgo-utils/mlgo/corpus/combine_training_corpus_lib.py",
+        "utils/mlgo-utils/mlgo/corpus/flags.py",
     ],
     imports = ["utils/mlgo-utils"],
 )