diff --git a/.github/workflows/build_torch_mlir.yml b/.github/workflows/build_torch_mlir.yml
index ccab990..82e72d6 100644
--- a/.github/workflows/build_torch_mlir.yml
+++ b/.github/workflows/build_torch_mlir.yml
@@ -131,7 +131,8 @@ jobs:
       - name: Patch/disable tests
         shell: bash
         run: |
-          sed -i.bak 's/add_subdirectory(test)/#add_subdirectory(test)/g' $TORCH_MLIR_MAIN_SRC_DIR/CMakeLists.txt
+          sed -i.bak 's/check-torch-mlir-capi/#check-torch-mlir-capi/g' $TORCH_MLIR_MAIN_SRC_DIR/CMakeLists.txt
+          sed -i.bak 's/add_subdirectory(CAPI)/#add_subdirectory(CAPI)/g' $TORCH_MLIR_MAIN_SRC_DIR/test/CAPI/CMakeLists.txt
 
       - name: Configure CMake
         run: |