diff --git a/extended/code_update/host_code_update.robot b/extended/code_update/host_code_update.robot
index 44f61af..cccb916 100644
--- a/extended/code_update/host_code_update.robot
+++ b/extended/code_update/host_code_update.robot
@@ -56,12 +56,12 @@
 Post Update Boot To OS
     [Documentation]  Boot the host OS
     [Tags]  Post_Update_Boot_To_OS
+    [Setup]  Start SOL Console Logging
     [Teardown]  Run Keywords  Stop SOL Console Logging
     ...         AND  Code Update Test Teardown
 
     Run Keyword If  '${PREV_TEST_STATUS}' == 'FAIL'
     ...  Fail  Code update failed. No need to boot to OS.
-    Start SOL Console Logging
     Delete Error Logs
     REST Power On
     Verify Running Host Image  ${IMAGE_FILE_PATH}