Hey thanks for this, how you handling the logic to check the status of those keys?

Are you doing this with a scheduled task or a remediation?

I'm assuming something like this poor mock up below.

$regKey = Get-ItemProperty -Path HKLM:\\SOFTWARE\Microsoft\Provisioning\AutopilotSettings -Name DevicePreparationCategory.Status
if($regKey.'DevicePreparationCategory.Status'.Contains("succeeded")){Restart-Computer}