diff --git a/src/Dialect/ONNX/BUILD b/src/Dialect/ONNX/BUILD
index f0b36b4..1144021 100644
--- a/src/Dialect/ONNX/BUILD
+++ b/src/Dialect/ONNX/BUILD
@@ -15,6 +15,9 @@ gentbl(
     td_includes = ["."],
     td_srcs = [
         "ONNXOps.td.inc",
+        # "//src/Interface:PromotableConstOperandsOpInterface.td",
+        # "//src/Interface:ResultTypeInferenceOpInterface.td",
+        # "//src/Interface:ShapeInferenceInterface.td",
         "@llvm-project//mlir:TdFiles",
     ],
 )
@@ -32,45 +35,10 @@ cc_library(
     deps = [
         ":OMONNXOpsIncGen",
         "//src/Interface:OMPromotableConstOperandsOpInterface",
-        "//src/Interface:OMShapeInferenceOpInterface",
         "//src/Interface:OMResultTypeInferenceOpInterface",
-        "@onnx",
-        # "@onnx//:onnx_proto_genproto",
-        # "@llvm-project//mlir:Affine",
-        "@llvm-project//mlir:IR",
-        "@llvm-project//mlir:Shape",
-
-        # "@llvm-project//mlir:AffineOps",
-        "@llvm-project//mlir:Dialect",
-        "@llvm-project//mlir:ExecutionEngine",
-        "@llvm-project//mlir:LLVMDialect",
-        # "@llvm-project//mlir:LoopOps",
+        "//src/Interface:OMShapeInferenceOpInterface",
+        "@llvm-project//mlir:AllTranslations",
         "@llvm-project//mlir:MlirOptLib",
-        "@llvm-project//mlir:Parser",
-        "@llvm-project//mlir:StandardOps",
-        "@llvm-project//mlir:Support",
-        "@llvm-project//mlir:VectorOps",
-
-        # "@llvm-project//mlir:Support",
-        # "@llvm-project//mlir:TableGen",
+        "@onnx",
     ],
 )
-
-# "@llvm-project//llvm:core",
-# "@llvm-project//llvm:dtu_asm_parser",
-# "@llvm-project//llvm:dtu_code_gen",
-# "@llvm-project//llvm:dtu_desc",
-# "@llvm-project//llvm:orc_jit",
-# "@llvm-project//llvm:passes",
-# "@llvm-project//llvm:x86_asm_parser",
-# "@llvm-project//llvm:x86_code_gen",
-# "@llvm-project//mlir:AffineOps",
-# "@llvm-project//mlir:Dialect",
-# "@llvm-project//mlir:ExecutionEngine",
-# "@llvm-project//mlir:LLVMDialect",
-# "@llvm-project//mlir:LoopOps",
-# "@llvm-project//mlir:MlirOptLib",
-# "@llvm-project//mlir:Parser",
-# "@llvm-project//mlir:StandardOps",
-# "@llvm-project//mlir:Support",
-# "@llvm-project//mlir:VectorOps",