diff --git a/extended/code_update/code_update_utils.robot b/extended/code_update/code_update_utils.robot
index d517f78..dac252b 100644
--- a/extended/code_update/code_update_utils.robot
+++ b/extended/code_update/code_update_utils.robot
@@ -48,8 +48,9 @@
     ${resp}=    openbmc post request    ${BMC_PREP_METHOD}   data=${data}
     should be equal as strings   ${resp.status_code}   ${HTTP_OK}
 
-    ${data}=      Read Properties     ${BMC_UPD_ATTR}
-    should contain    ${data['status']}   Switch to update mode in progress
+    # Update method will reset the BMC, adding delay for reboot to
+    # come into force.
+    Sleep  10s
 
 
 SCP Tar Image File to BMC