diff --git a/scripts/Windows/OS/Upgrade.ps1 b/scripts/Windows/OS/Upgrade.ps1
index a7930b33..37df467d 100644
--- a/scripts/Windows/OS/Upgrade.ps1
+++ b/scripts/Windows/OS/Upgrade.ps1
@@ -1,3 +1,5 @@
+. "$PSScriptRoot/../Scripts/Context.ps1";
+
 function Invoke-Upgrade([Context] $context)
 {
     Write-Host "Upgrading Windows";