diff --git a/.github/workflows/precompiled_nif.yml b/.github/workflows/precompiled_nif.yml
index 7de01cd..daf49fb 100644
--- a/.github/workflows/precompiled_nif.yml
+++ b/.github/workflows/precompiled_nif.yml
@@ -38,7 +38,6 @@ jobs:
           mix local.hex --force
           mix local.rebar --force
           echo "NX_IREE_SOURCE_DIR=$(elixir -e 'IO.puts(Path.absname(~s(${{env.NX_IREE_SOURCE_DIR}})))')" > $GITHUB_ENV
-          echo $NX_IREE_SOURCE_DIR
 
       - name: Cache mix dependencies
         id: cache-mix-deps
diff --git a/cmake/CMakeLists.txt b/cmake/CMakeLists.txt
index b510740..37d46a0 100644
--- a/cmake/CMakeLists.txt
+++ b/cmake/CMakeLists.txt
@@ -28,7 +28,7 @@ add_library(${_NAME} SHARED ${iree_runtime_sources})
 
 set(__BUILD_DIR "${IREE_RUNTIME_BUILD_DIR}")
 
-add_subdirectory("${IREE_DIR}" ${__BUILD_DIR} EXCLUDE_FROM_ALL)
+add_subdirectory("${NX_IREE_SOURCE_DIR}" ${__BUILD_DIR} EXCLUDE_FROM_ALL)
 
 install(
     TARGETS ${_NAME}
diff --git a/mix.exs b/mix.exs
index f694120..a800039 100644
--- a/mix.exs
+++ b/mix.exs
@@ -24,7 +24,7 @@ defmodule NxIREE.MixProject do
           "CWD_RELATIVE_TO_PRIV_PATH" => cwd_relative_to_priv,
           "MAKE_NUM_JOBS" => n_jobs,
           "IREE_GIT_REV" => nx_iree_config().tag,
-          "IREE_DIR" => nx_iree_config().source_dir
+          "NX_IREE_SOURCE_DIR" => nx_iree_config().source_dir
         }
       end
     ]