diff --git a/Update-VMGpuPartitionDriver.ps1 b/Update-VMGpuPartitionDriver.ps1
index 5215258..a74a8ea 100644
--- a/Update-VMGpuPartitionDriver.ps1
+++ b/Update-VMGpuPartitionDriver.ps1
@@ -67,7 +67,7 @@ foreach ($d in $drivers) {
 
 }
 
-$VM = Get-VM -VMName "GPUP"
+$VM = Get-VM -VMName $VMName
 $VHD = Get-VHD -VMId $VM.VMId
 
 If ($VM.state -eq "Running") {