diff --git a/scripts/Windows/Scripts/PowerManagement.ps1 b/scripts/Windows/Scripts/PowerManagement.ps1
index 39305a94..f29410a3 100644
--- a/scripts/Windows/Scripts/PowerManagement.ps1
+++ b/scripts/Windows/Scripts/PowerManagement.ps1
@@ -153,12 +153,15 @@ $null = New-Module {
     #>
     function Restart-Intermediate {
         param(
-            [switch] $DefaultUser
+            [Parameter(ParameterSetName="Default", Mandatory)]
+            [switch] $DefaultUser,
+            [Parameter(ParameterSetName="Current", Mandatory)]
+            [switch] $CurrentUser
         )
 
         $register = { param($UserKey) Register-Setup @PSBoundParameters; };
 
-        if ($DefaultUser) {
+        if ($DefaultUser.ToBool()) {
             Edit-DefaultUserKey {
                 param(
                     [RegistryKey] $Key
@@ -166,6 +169,8 @@ $null = New-Module {
 
                 & $register $Key;
             }
+        } elseif ($CurrentUser.ToBool()) {
+            & $register (Get-Item HKCU:\);
         } else {
             & $register;
         }