diff --git a/.github/scripts/setup_env.bash b/.github/scripts/setup_env.bash
index cc26a12d3..358371b2f 100755
--- a/.github/scripts/setup_env.bash
+++ b/.github/scripts/setup_env.bash
@@ -24,6 +24,8 @@
 # shellcheck disable=SC1091,SC2128
 . "$( dirname -- "$BASH_SOURCE"; )/utils_triton.bash"
 # shellcheck disable=SC1091,SC2128
+. "$( dirname -- "$BASH_SOURCE"; )/utils_torchrec.bash"
+# shellcheck disable=SC1091,SC2128
 . "$( dirname -- "$BASH_SOURCE"; )/fbgemm_build.bash"
 # shellcheck disable=SC1091,SC2128
 . "$( dirname -- "$BASH_SOURCE"; )/fbgemm_gpu_build.bash"
diff --git a/.github/scripts/utils_torchrec.bash b/.github/scripts/utils_torchrec.bash
index a65f6bd75..39fd4d3f9 100644
--- a/.github/scripts/utils_torchrec.bash
+++ b/.github/scripts/utils_torchrec.bash
@@ -45,7 +45,7 @@ install_torchrec_pip () {
   (exec_with_retries 3 conda install ${env_prefix} -c conda-forge -y \
     iopath \
     lightning-utilities \
-    pyre_extensions) || return 1
+    pyre-extensions) || return 1
 
   # Install the package from TorchRec PIP (not PyPI)
   install_from_pytorch_pip "${env_name}" torchrec "${torchrec_channel_version}" "${torchrec_variant_type_version}" || return 1