diff --git a/.github/workflows/e2e-tests.yml b/.github/workflows/e2e-tests.yml
index 4dbb8079..ce493934 100644
--- a/.github/workflows/e2e-tests.yml
+++ b/.github/workflows/e2e-tests.yml
@@ -34,8 +34,7 @@ jobs:
           target-repo: wasp-lang/wasp
 
       - name: Download Wasp build artifacts from linked PR
-        # TODO: Change to `main` branch once https://github.com/wasp-lang/wasp/pull/3335 is merged.
-        uses: wasp-lang/wasp/.github/actions/fetch-nightly-cli@cprecioso/add-error-to-action
+        uses: wasp-lang/wasp/.github/actions/fetch-nightly-cli@main
         with:
           output-dir: ${{ runner.temp }}/wasp-cli
           branch: ${{ steps.get-linked-pr-branch.outputs.pr-branch || 'main' }}