1 files changed, 10 insertions, 2 deletions
diff --git a/spec/features/profiles/gpg_keys_spec.rb b/spec/features/profiles/gpg_keys_spec.rb
index f39d9ddaf56..38abf9d20b0 100644
--- a/spec/features/profiles/gpg_keys_spec.rb
+++ b/spec/features/profiles/gpg_keys_spec.rb
@@ -15,6 +15,7 @@ RSpec.describe 'Profile > GPG Keys', feature_category: :user_profile do
     end
 
     it 'saves the new key' do
+      click_button('Add new key')
       fill_in('Key', with: GpgHelpers::User2.public_key)
       click_button('Add key')
 
@@ -24,6 +25,7 @@ RSpec.describe 'Profile > GPG Keys', feature_category: :user_profile do
     end
 
     it 'with multiple subkeys' do
+      click_button('Add new key')
       fill_in('Key', with: GpgHelpers::User3.public_key)
       click_button('Add key')
 
@@ -52,7 +54,10 @@ RSpec.describe 'Profile > GPG Keys', feature_category: :user_profile do
 
     click_link('Remove')
 
-    expect(page).to have_content('Your GPG keys (0)')
+    expect(page).to have_content('Your GPG keys')
+    page.within('.gl-new-card-count') do
+      expect(page).to have_content('0')
+    end
   end
 
   it 'user revokes a key via the key index' do
@@ -63,7 +68,10 @@ RSpec.describe 'Profile > GPG Keys', feature_category: :user_profile do
 
     click_link('Revoke')
 
-    expect(page).to have_content('Your GPG keys (0)')
+    expect(page).to have_content('Your GPG keys')
+    page.within('.gl-new-card-count') do
+      expect(page).to have_content('0')
+    end
 
     expect(gpg_signature.reload).to have_attributes(
       verification_status: 'unknown_key',