diff --git a/lib/redfish_code_update_utils.robot b/lib/redfish_code_update_utils.robot
index cf0ae51..0b99ca4 100644
--- a/lib/redfish_code_update_utils.robot
+++ b/lib/redfish_code_update_utils.robot
@@ -108,7 +108,7 @@
     [Arguments]  ${apply_time}
 
     # Description of argument(s):
-    # policy     ApplyTime allowed values (e.g. "OnReset", "Immediate").
+    # apply_time     ApplyTime allowed values (e.g. "OnReset", "Immediate").
 
     Set ApplyTime  policy=${apply_Time}
     Redfish Upload Image  ${REDFISH_BASE_URI}UpdateService  ${IMAGE_FILE_PATH}
@@ -135,7 +135,7 @@
     [Arguments]  ${apply_time}  ${start_boot_seconds}
 
     # Description of argument(s):
-    # policy                ApplyTime allowed values
+    # apply_time            ApplyTime allowed values
     #                       (e.g. "OnReset", "Immediate").
     # start_boot_seconds    See 'Wait For Reboot' for details.
 
@@ -151,8 +151,16 @@
 
 Poweron Host And Verify Host Image
     [Documentation]  Power on Host and verify installed image is functional.
+    [Arguments]  ${apply_time}
 
-    Redfish Power On
+    # Description of argument(s):
+    # apply_time            ApplyTime allowed values
+    #                       (e.g. "OnReset", "Immediate").
+
+    Run Keyword if  'OnReset' == '${apply_time}'
+    ...    Redfish Host Reboot
+    ...  ELSE
+    ...    Wait State  os_running_match_state  10 mins
     Redfish.Login
     Redfish Verify Host Version  ${IMAGE_FILE_PATH}