diff --git a/src/Dialect/ONNX/BUILD b/src/Dialect/ONNX/BUILD
index 535f382..ca5e775 100644
--- a/src/Dialect/ONNX/BUILD
+++ b/src/Dialect/ONNX/BUILD
@@ -12,14 +12,13 @@ gentbl(
     ],
     tblgen = "@llvm-project//mlir:mlir-tblgen",
     td_file = "ONNXOps.td",
-    td_includes = ["."],
-    # td_relative_includes = ["../../"]
+    td_includes = [
+        ".",
+        "external/onnx-mlir",
+    ],
     td_srcs = [
         "ONNXOps.td.inc",
         "//src/Interface:TdFiles",
-        # "//src/Interface:PromotableConstOperandsOpInterface.td",
-        # "//src/Interface:ResultTypeInferenceOpInterface.td",
-        # "//src/Interface:ShapeInferenceInterface.td",
         "@llvm-project//mlir:TdFiles",
     ],
 )