diff --git a/lib/state_manager.robot b/lib/state_manager.robot
index a780631..a9082f8 100755
--- a/lib/state_manager.robot
+++ b/lib/state_manager.robot
@@ -115,10 +115,6 @@
     #    "RequestedHostTransition": "xyz.openbmc_project.State.Host.Transition.On"
     # }
 
-    # TODO: Remove this logic once migration is complete.
-    ${status}=  Run Keyword And Return Status  Is OS Starting
-    Return From Keyword If   '${status}' == '${True}'    ${True}
-
     ${boot_stage}=  Get Host State Attribute  BootProgress
     Should Be Equal  ${OS_BOOT_START}  ${boot_stage}