diff --git a/TESTS/host_tests/unexpected_reset.py b/TESTS/host_tests/unexpected_reset.py
index 2df1ef3605f..e85f3d10f01 100644
--- a/TESTS/host_tests/unexpected_reset.py
+++ b/TESTS/host_tests/unexpected_reset.py
@@ -49,7 +49,7 @@ def run_gen(key, value, time):
             except StopIteration:
                 self._error = True
 
-        for resp in ("start", "read", "format_done", "soft_reset_dut_complete"):
+        for resp in ("start", "read", "format_done", "reset_complete"):
             self.register_callback(resp, run_gen)
 
     def teardown(self):
@@ -86,7 +86,7 @@ def unexpected_reset_test(self):
             # Wait for start token
             key, value, time = yield
             self.log("Key from yield: %s" % key)
-            if key != "soft_reset_dut_complete":
+            if key != "reset_complete":
                 return