diff --git a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
index 43613ad507070..be18ba5b46806 100644
--- a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
@@ -1029,6 +1029,7 @@ gentbl_filegroup(
     td_file = "mlir/dialects/ShardOps.td",
     deps = [
         "//mlir:OpBaseTdFiles",
+        "//mlir:ShapeOpsTdFiles",
         "//mlir:ShardTdFiles",
     ],
 )