diff --git a/.github/actions/docker-run/action.yml b/.github/actions/docker-run/action.yml
index 59a26859289..2977136d126 100644
--- a/.github/actions/docker-run/action.yml
+++ b/.github/actions/docker-run/action.yml
@@ -80,7 +80,7 @@ runs:
           ${{ inputs.device }}
           -w ${{ github.workspace }}
         run: |
-          if [ ${{ inputs.install_wheel }} ]; then
+          if [ ${{ inputs.install_wheel }} = "true" ]; then
             WHEEL_FILENAME=$(ls -1 *.whl)
             pip3 install $WHEEL_FILENAME
           fi