diff --git a/.bazelversion b/.bazelversion
new file mode 100644
index 0000000..fd2a018
--- /dev/null
+++ b/.bazelversion
@@ -0,0 +1 @@
+3.1.0
diff --git a/BUILD b/BUILD
index e21d530..8db4ec8 100644
--- a/BUILD
+++ b/BUILD
@@ -8,10 +8,10 @@ cc_library(
     includes = [
         "include"
     ],
-    hdrs = 
+    hdrs =
         glob(["include/tim/utils/nbg_parser/*.h"])
     ,
-    srcs = 
+    srcs =
         glob(["src/tim/utils/nbg_parser/*.c"])
     ,
     linkstatic = True,
@@ -24,6 +24,7 @@ cc_library(
     includes = [
         "include",
         "src/tim/vx",
+        "src/tim/transform",
     ],
     hdrs = [
         "include/tim/vx/context.h",
@@ -31,7 +32,7 @@ cc_library(
         "include/tim/vx/operation.h",
         "include/tim/vx/tensor.h",
         "include/tim/vx/types.h",
-        "include/tim/transform/layout_inference.h"
+        "include/tim/transform/layout_inference.h",
     ] + glob([
         "include/tim/vx/ops/*.h"
     ]),