diff --git a/extended/code_update/update_bmc.robot b/extended/code_update/update_bmc.robot
index 6d4ca55..b2d7692 100644
--- a/extended/code_update/update_bmc.robot
+++ b/extended/code_update/update_bmc.robot
@@ -16,6 +16,7 @@
 ...                 - Version check post update
 
 Resource          code_update_utils.robot
+Resource          ../../lib/boot/boot_resource_master.robot
 
 *** Variables ***
 
@@ -28,7 +29,13 @@
 
     Check If File Exist    ${FILE_PATH}
     System Readiness Test
-    Validate BMC Version   before
+    ${status}=   Run Keyword and Return Status
+    ...   Validate BMC Version   before
+
+    Run Keyword if  '${status}' == '${False}'
+    ...     Pass Execution   Same Driver version installed
+
+    BMC Power Off
 
     Preserve BMC Network Setting
     SCP Tar Image File to BMC   ${FILE_PATH}