diff --git a/testing/integration/upgrade_test.go b/testing/integration/upgrade_test.go
index ebd27881f9b..e17938b3d1a 100644
--- a/testing/integration/upgrade_test.go
+++ b/testing/integration/upgrade_test.go
@@ -1186,7 +1186,7 @@ func TestStandaloneUpgradeFailsRestart(t *testing.T) {
 	}
 
 	// Ensure that the Upgrade Watcher has stopped running.
-	checkUpgradeWatcherRan(t, fromF, fromVersionParsed)
+	waitForUpgradeWatcherToComplete(t, fromF, fromVersionParsed, standaloneWatcherDuration)
 
 	// Ensure that the original version of Agent is running again.
 	t.Log("Check Agent version to ensure rollback is successful")