diff --git a/features/steps/project/project.rb b/features/steps/project/project.rb
index 98b57e5cbfbe21766426c3854520d6544a487bf0..76fefee9254f12cae6041e1aba43654b12f143d4 100644
--- a/features/steps/project/project.rb
+++ b/features/steps/project/project.rb
@@ -134,8 +134,8 @@ class Spinach::Features::Project < Spinach::FeatureSteps
   end
 
   step 'I should see Notification saved message' do
-    page.within '.flash-container' do
-      expect(page).to have_content 'Notification settings saved'
+    page.within '#notifications-button' do
+      expect(page).to have_content 'On mention'
     end
   end