diff --git a/Sources/Winget-AutoUpdate/functions/Update-WAU.ps1 b/Sources/Winget-AutoUpdate/functions/Update-WAU.ps1
index 86e870d..e39dfc9 100644
--- a/Sources/Winget-AutoUpdate/functions/Update-WAU.ps1
+++ b/Sources/Winget-AutoUpdate/functions/Update-WAU.ps1
@@ -96,13 +96,16 @@ function Update-WAU {
                 Copy-Item -Path "$TempPath\*" -Destination "$WorkingDir\" -Exclude "icons" -Recurse -Force
             }
 
+            #Get installed version
+            $InstalledVersion = Get-Content "$TempPath\Version.txt"
+
             #Remove update zip file and update temp folder
             Write-ToLog "Done. Cleaning temp files..." "Cyan"
             Remove-Item -Path $ZipFile -Force -ErrorAction SilentlyContinue
             Remove-Item -Path $location -Recurse -Force -ErrorAction SilentlyContinue
 
             #Set new version to registry
-            New-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\Winget-AutoUpdate" -Name "DisplayVersion" -Value $WAUAvailableVersion -Force | Out-Null
+            New-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\Winget-AutoUpdate" -Name "DisplayVersion" -Value $InstalledVersion -Force | Out-Null
 
             #Set Post Update actions to 1
             New-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\Winget-AutoUpdate" -Name "WAU_PostUpdateActions" -Value 1 -Force | Out-Null