diff --git a/extended/code_update/code_update.robot b/extended/code_update/code_update.robot
index 470ba13..5398d7e 100644
--- a/extended/code_update/code_update.robot
+++ b/extended/code_update/code_update.robot
@@ -70,10 +70,7 @@
     ${software_state}=  Read Properties  ${SOFTWARE_VERSION_URI}${version_id}
     Should Be Equal As Strings  &{software_state}[Activation]  ${ACTIVE}
 
-    # TODO: Switch OBMC Reboot (off) once it's fixed
-    Trigger Warm Reset Via Reboot
-    Check If BMC is Up
-    Wait For BMC Ready
+    OBMC Reboot (off)
 
 
 Post Update Boot To OS