diff --git a/action.yml b/action.yml
index 201a8c9..990de1d 100644
--- a/action.yml
+++ b/action.yml
@@ -14,12 +14,11 @@ runs:
       run: |
         set -x
         if [[ "${{ runner.os }}" == "Linux" ]]; then
-          cache_dir=$(apt-cache policy postgresql-${{ inputs.postgresql_version }} | awk '/Filename/ {print $2}')
+          cache_dir="$(apt-cache policy postgresql-${{ inputs.postgresql_version }} | awk '/Filename/ {print $2}')"
         elif [[ "${{ runner.os }}" == "macOS" ]]; then
-          cache_dir=$(brew --cache postgresql@${{ inputs.postgresql_version }})
+          cache_dir="$(brew --cache postgresql@${{ inputs.postgresql_version }})"
         elif [[ "${{ runner.os }}" == "Windows" ]]; then
-          choco search postgresql --download-cache
-          cache_dir=$(choco cache list | grep 'postgresql${{ inputs.postgresql_version }}')
+          cache_dir="~/AppData/Local/Temp/chocolatey/postgresql-${{ inputs.postgresql_version }}"
         fi
         echo "cache_dir=$cache_dir" >> "$GITHUB_OUTPUT"
       shell: bash