diff --git a/.github/actions/test/action.yml b/.github/actions/test/action.yml
index cc892944ea8..28aa53086bb 100644
--- a/.github/actions/test/action.yml
+++ b/.github/actions/test/action.yml
@@ -99,12 +99,12 @@ runs:
         Initialize-Disk -Number $diskNumber -PartitionStyle MBR
 
         New-Partition -DiskNumber $diskNumber -UseMaximumSize -AssignDriveLetter
-        Get-Partition -DiskNumber 2 | Set-Partition -NewDriveLetter Z
+        Get-Partition -DiskNumber $diskNumber | Set-Partition -NewDriveLetter Z
 
         # Output the details of the new drive
         Write-Output "New virtual disk created and accessible at drive letter: Z"
 
-        echo "TARGET_DRIVE=Z:\\" >> %GITHUB_ENV%
+        echo "TARGET_DRIVE=Microsoft Virtual Disk" >> %GITHUB_ENV%
 
     - name: Test release
       shell: bash
@@ -125,7 +125,7 @@ runs:
         if [[ '${{ runner.os }}' == 'Linux' ]]; then
           # linux doesn't support e2e tests yet
           npm run wdio:unit
-        elif [[ '${{ runner.os }}' == 'Windows' ]]; then
+        else
           # download the test source
           wget -q -O ${{ env.TEST_SOURCE_FILE }} ${{ env.TEST_SOURCE_URL }}
           # E2E tests can't input the administrative password, therefore the tests need to run as root