diff --git a/extended/code_update/update_bmc.robot b/extended/code_update/update_bmc.robot
index dbdd7d1..b3b05c1 100644
--- a/extended/code_update/update_bmc.robot
+++ b/extended/code_update/update_bmc.robot
@@ -8,7 +8,7 @@
 ...               Update work flow sequence:
 ...                 - User input BMC File existence check
 ...                 - Ping Test and REST authentication
-...                 - Set Host Power host setting Policy to RESTORE_LAST_STATE
+...                 - Set Host Power host setting Policy to ALWAYS_POWER_OFF
 ...                   On reboot this policy would ensure the BMC comes
 ...                   online and stays at HOST_POWERED_OFF state.
 ...                 - Issue poweroff
@@ -82,7 +82,7 @@
     Check Boot Count And Time
     Prune Journal Log
     Power Off Request
-    Set BMC Power Policy  RESTORE_LAST_STATE
+    Set BMC Power Policy  ALWAYS_POWER_OFF
 
     Prepare For Update
     Check If BMC is Up  20 min  10 sec
diff --git a/extended/standby_bmc.robot b/extended/standby_bmc.robot
index 988b6d9..9a5bdb9 100644
--- a/extended/standby_bmc.robot
+++ b/extended/standby_bmc.robot
@@ -7,7 +7,7 @@
 ...                  - Power state is 0 (off)
 ...                  - BMC state is "Ready"
 ...                  - HOST state is "Off"
-...                  - Boot policy is "RESTORE_LAST_STATE"
+...                  - Boot policy is "ALWAYS_POWER_OFF"
 ...               Power cycle system via PDU if specified
 ...               Prune archived journal logs
 
@@ -53,7 +53,7 @@
 
     Prune Journal Log
 
-    Run Keyword And Ignore Error  Set BMC Power Policy  ${RESTORE_LAST_STATE}
+    Run Keyword And Ignore Error  Set BMC Power Policy  ${ALWAYS_POWER_OFF}
 
     # TODO: Enable MAC AES check latter.
     # Reference : openbmc/openbmc-test-automation#998