diff --git a/regression-test/suites/backup_restore/test_backup_cancelled.groovy b/regression-test/suites/backup_restore/test_backup_cancelled.groovy
index ee8e9a2d5e24b0..a5db3bc9fd3539 100644
--- a/regression-test/suites/backup_restore/test_backup_cancelled.groovy
+++ b/regression-test/suites/backup_restore/test_backup_cancelled.groovy
@@ -110,9 +110,7 @@ suite("test_backup_cancelled", "backup_cancelled") {
         )
     """
 
-    while (syncer.checkSnapshotFinish(dbName) == false) {
-        Thread.sleep(3000)
-    }
+    syncer.waitAllRestoreFinish(dbName)
 
     result = sql "SELECT * FROM ${dbName}.${tableName}"
     assertEquals(result.size(), values.size());