diff --git a/.github/actions/setup_env/action.yml b/.github/actions/setup_env/action.yml
index 6052d6385f6..8c64f24dff4 100644
--- a/.github/actions/setup_env/action.yml
+++ b/.github/actions/setup_env/action.yml
@@ -10,24 +10,23 @@ inputs:
 runs:
   using: "composite"
   steps:
-      # - name: Download Lock File
-      #   run:  wget -q https://raw.githubusercontent.com/tardis-sn/tardis/master/conda-${{ inputs.os-label }}.lock
-      #   if: matrix.pip == true
-      #   shell: bash
+      - name: Download Lock File
+        run:  wget -q https://raw.githubusercontent.com/tardis-sn/tardis/master/conda-${{ inputs.os-label }}.lock
+        if: matrix.pip == true
+        shell: bash
       
-      # - name: Generate Cache Key
-      #   run: | 
-      #     file_hash=$(cat conda-${{ inputs.os-label }}.lock | shasum -a 256 | cut -d' ' -f1)
-      #     echo "file_hash=$file_hash" >> "${GITHUB_OUTPUT}"
-      #   id: cache-environment-key
-      #   shell: bash
+      - name: Generate Cache Key
+        run: | 
+          file_hash=$(cat conda-${{ inputs.os-label }}.lock | shasum -a 256 | cut -d' ' -f1)
+          echo "file_hash=$file_hash" >> "${GITHUB_OUTPUT}"
+        id: cache-environment-key
+        shell: bash
         
       - uses: mamba-org/setup-micromamba@v1
         with:
           environment-file: conda-${{ inputs.os-label }}.lock
-          # cache-environment-key: ${{ steps.cache-environment-key.outputs.file_hash }}
-          # cache-downloads-key: ${{ steps.cache-environment-key.outputs.file_hash }}
+          cache-environment-key: ${{ steps.cache-environment-key.outputs.file_hash }}
+          cache-downloads-key: ${{ steps.cache-environment-key.outputs.file_hash }}
           environment-name: tardis
-          # cache-environment: true
-          # cache-downloads: true
-
+          cache-environment: true
+          cache-downloads: true