diff --git a/tests/test_association.robot b/tests/test_association.robot
index dacf0a1..28809ad 100755
--- a/tests/test_association.robot
+++ b/tests/test_association.robot
@@ -5,6 +5,7 @@
 Resource          ../lib/utils.robot
 Resource          ../lib/connection_client.robot
 Resource          ../lib/openbmc_ffdc.robot
+Resource          ../lib/boot/boot_resource_master.robot
 
 Library           Collections
 
@@ -214,6 +215,9 @@
     ${pre_reboot_association_content} =
     ...   Read Attribute   ${association_uri}    endpoints
 
+    Initiate Power Off
+    Check Power Off States
+
     ${output}=      Execute Command    /sbin/reboot
     Check If BMC is Up   5 min    10 sec
 
diff --git a/tests/test_eventlog.robot b/tests/test_eventlog.robot
index d2eadc4..5bb59f6 100644
--- a/tests/test_eventlog.robot
+++ b/tests/test_eventlog.robot
@@ -6,6 +6,7 @@
 Resource          ../lib/utils.robot
 Resource          ../lib/connection_client.robot
 Resource          ../lib/openbmc_ffdc.robot
+Resource          ../lib/boot/boot_resource_master.robot
 
 Library           Collections
 
@@ -153,6 +154,7 @@
     ...                 openbmc reboot
     ...                 Steps:
     ...                     Create event,
+    ...                     Power off if ON else no-op
     ...                     Reboot openbmc,
     ...                     Wait for BMC to READY or Powered OFF state
     ...                     Events should exist post reboot,
@@ -161,6 +163,9 @@
     [Tags]      bmcreboot
     ${pre_reboot_event}=         create a test log
 
+    Initiate Power Off
+    Check Power Off States
+
     ${output}=      Execute Command    /sbin/reboot
     Check If BMC is Up   5 min    10 sec