diff --git a/tests/codeception/acceptance/update-wordpoints.Cept.php b/tests/codeception/acceptance/update-wordpoints.Cept.php
index 694e8c4..f2e2a73 100644
--- a/tests/codeception/acceptance/update-wordpoints.Cept.php
+++ b/tests/codeception/acceptance/update-wordpoints.Cept.php
@@ -21,6 +21,7 @@
 $I->click( 'Install Update Now' );
 $I->switchToIFrame();
 $I->waitForJqueryAjax( 30 );
+$I->waitForElement( '.updated-message.notice-success' );
 $I->see( 'Updated!' );
 
 // EOF