diff --git a/extended/code_update/code_update.robot b/extended/code_update/code_update.robot
index d2e16ed..1d3304f 100644
--- a/extended/code_update/code_update.robot
+++ b/extended/code_update/code_update.robot
@@ -22,6 +22,7 @@
 Resource          ../lib/rest_client.robot
 Resource          ../lib/openbmc_ffdc.robot
 Resource          ../../lib/code_update_utils.robot
+Resource          ../../lib/boot_utils.robot
 
 Test Teardown     Code Update Teardown
 
@@ -69,6 +70,17 @@
     ${software_state}=  Read Properties  ${SOFTWARE_VERSION_URI}${version_id}
     Should Be Equal As Strings  &{software_state}[Activation]  ${ACTIVE}
 
+    OBMC Reboot (off)
+
+
+Post Update Boot To OS
+    [Documentation]  Boot the host OS
+    [Tags]  Post_Update_Boot_To_OS
+
+    Run Keyword Unless  '${PREV_TEST_STATUS}' == 'PASS'
+    ...  Fail  Code update failed. No need to boot to OS.
+    REST Power On
+
 
 Host Image Priority Attribute Test
     [Documentation]  Set "Priority" attribute.
@@ -80,6 +92,7 @@
     Priority          ${1}
     Priority          ${127}
 
+
 *** Keywords ***
 
 Set PNOR Attribute