diff --git a/tests/common/reboot.py b/tests/common/reboot.py
index 09183f7cb6c..1c1cbd50515 100644
--- a/tests/common/reboot.py
+++ b/tests/common/reboot.py
@@ -161,7 +161,7 @@ def execute_reboot_helper():
         logger.info('waiting for warmboot-finalizer service to become activating')
         finalizer_state = get_warmboot_finalizer_state(duthost)
         while finalizer_state != 'activating':
-            dut_datetime_after_ssh = duthost.get_up_time()
+            dut_datetime_after_ssh = duthost.get_now_time()
             time_passed = float(dut_datetime_after_ssh.strftime("%s")) - float(dut_datetime.strftime("%s"))
             if time_passed > wait:
                 raise Exception('warmboot-finalizer never reached state "activating"')