diff --git a/extended/code_update/bmc_code_update.robot b/extended/code_update/bmc_code_update.robot
index 7a92e00..bdcfdd5 100644
--- a/extended/code_update/bmc_code_update.robot
+++ b/extended/code_update/bmc_code_update.robot
@@ -73,6 +73,7 @@
     [Teardown]  REST BMC Code Update Teardown
 
     Run Keyword And Ignore Error  List Installed Images  BMC
+    Run Keyword And Ignore Error  Set ApplyTime  policy=OnReset
 
     ${image_version}=  Get Version Tar  ${IMAGE_FILE_PATH}
     ${bmc_release_info}=  Get BMC Release Info