diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 7ab1fb2..54f0f05 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -26,11 +26,9 @@ jobs:
         run: |
           wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.3.1%2Bcpu.zip
           unzip libtorch-cxx11-abi-shared-with-deps-2.3.1+cpu.zip
-          export LIBTORCH=`pwd`/libtorch
-          export Torch_DIR=`pwd`/libtorch
       - name: Tests & build-wrapper
         run: |
-          cmake -S . -B build -Wno-dev 
+          cmake -S . -B build -Wno-dev -DCMAKE_PREFIX_PATH=$(pwd)/libtorch
           build-wrapper-linux-x86-64 --out-dir ${{ env.BUILD_WRAPPER_OUT_DIR }} cmake --build build/ --config Release
           cd build
           make