diff --git a/build_tools/ci/run_matmul_test.sh b/build_tools/ci/run_matmul_test.sh
index bda8b753e..ac9ca962f 100755
--- a/build_tools/ci/run_matmul_test.sh
+++ b/build_tools/ci/run_matmul_test.sh
@@ -555,17 +555,16 @@ run_matmul_test \
 # MLIR-AIR Matmul tests
 ###################################################################
 
-# TODO: re-enable after fixing in AIR
-# if [ -d "$VITIS" ]; then
-#   run_matmul_test \
-#       --name_prefix "ukern" \
-#       --lower_to_aie_pipeline "air" \
-#       --tile_pipeline "pad-pack" \
-#       --lhs_rhs_type "bf16" \
-#       --acc_type "f32" \
-#       --m "256"  --k "256" --n "256" \
-#       --use_ukernel "1"
-# fi
+if [ -d "$VITIS" ]; then
+  run_matmul_test \
+      --name_prefix "ukern" \
+      --lower_to_aie_pipeline "air" \
+      --tile_pipeline "pad-pack" \
+      --lhs_rhs_type "bf16" \
+      --acc_type "f32" \
+      --m "256"  --k "256" --n "256" \
+      --use_ukernel "1"
+fi
 
 # Example of a run with a group of 2+ matmuls. Currently this test is passed
 # the flag '--num_repeat_runs 0" as there is currently an issue with the runtime if
@@ -593,6 +592,7 @@ run_matmul_test \
 
 run_matmul_test \
     --name_prefix "packPeel_i32" \
+    --lower_to_aie_pipeline "air" \
     --tile_pipeline "pack-peel" \
     --lhs_rhs_type "i32" \
     --acc_type "i32" \
@@ -600,6 +600,7 @@ run_matmul_test \
 
 run_matmul_test \
     --name_prefix "packPeel_bf16" \
+    --lower_to_aie_pipeline "air" \
     --tile_pipeline "pack-peel" \
     --lhs_rhs_type "bf16" \
     --acc_type "f32" \
@@ -607,6 +608,7 @@ run_matmul_test \
 
 run_matmul_test \
   --name_prefix "packPeel_t_bf16" \
+  --lower_to_aie_pipeline "air" \
   --tile_pipeline "pack-peel" \
   --lhs_rhs_type "bf16" \
   --acc_type "f32" \
diff --git a/third_party/mlir-air b/third_party/mlir-air
index 3d1a4e19f..e3a725381 160000
--- a/third_party/mlir-air
+++ b/third_party/mlir-air
@@ -1 +1 @@
-Subproject commit 3d1a4e19ff748897a37e0eb88b59999197dbc0f8
+Subproject commit e3a72538111219a78d2b98a9b6f074bb4d2e7cd7