diff --git a/rhodecode/apps/repository/tests/test_ai_code_review.py b/rhodecode/apps/repository/tests/test_ai_code_review.py
index 25c0d1bb..4c785091 100644
--- a/rhodecode/apps/repository/tests/test_ai_code_review.py
+++ b/rhodecode/apps/repository/tests/test_ai_code_review.py
@@ -53,7 +53,7 @@ class TestAiCodeReview:
 
             session = response.get_session_from_response()
             err_msg = session.peek_flash('error')
-            success_message = session.peek_flash()
+            success_message = session.peek_flash('success')
 
             assert response.status_code == 200
             assert response.json_body == {'response': True, 'redirect_url': redirect_url}
@@ -80,7 +80,12 @@ class TestAiCodeReview:
 
             ai_user = User.get_ai_user()
 
+            session = response.get_session_from_response()
+            success_message = session.peek_flash('success')
+
             assert response.status_code == 200
             assert run_task_mock.call_count == 1
             assert len(existing_reviewers) == 1
-            assert existing_reviewers[0].user_id == ai_user.user_id
\ No newline at end of file
+            assert existing_reviewers[0].user_id == ai_user.user_id
+            assert len(success_message) == 1
+            assert success_message[0] == "AI code review has started. The review will be ready in about 1–3 minutes."