diff --git a/extended/test_bios_update.robot b/extended/test_bios_update.robot
index dd0334f..887f48a 100644
--- a/extended/test_bios_update.robot
+++ b/extended/test_bios_update.robot
@@ -6,6 +6,7 @@
 Resource          ../lib/utils.robot
 Resource          ../lib/connection_client.robot
 Resource          ../lib/openbmc_ffdc.robot
+Resource          ../lib/state_manager.robot
 
 Test Teardown     FFDC On Test Case Fail
 
@@ -32,8 +33,7 @@
     Trigger Warm Reset
     Check If BMC is Up  20 min  10 sec
 
-    Wait Until Keyword Succeeds
-    ...  20 min  10 sec  Verify BMC State  BMC_READY
+    Wait For BMC Ready
 
     Clear BMC Record Log