diff --git a/windows/src/desktop/kmshell/main/Keyman.System.UpdateStateMachine.pas b/windows/src/desktop/kmshell/main/Keyman.System.UpdateStateMachine.pas
index 80fbf08634..c77510af6c 100644
--- a/windows/src/desktop/kmshell/main/Keyman.System.UpdateStateMachine.pas
+++ b/windows/src/desktop/kmshell/main/Keyman.System.UpdateStateMachine.pas
@@ -567,7 +567,7 @@ function TUpdateStateMachine.ReadyToInstall: Boolean;
 begin
   if not IsCurrentStateAssigned then
     Exit(False);
-  if (CurrentState.ClassName = 'WaitingRestartState') and not HasKeymanRun then
+  if (CurrentState is WaitingRestartState) and not HasKeymanRun then
     Result := True
   else
     Result := False;