1 files changed, 4 insertions, 6 deletions
diff --git a/spec/frontend/groups/components/invite_members_banner_spec.js b/spec/frontend/groups/components/invite_members_banner_spec.js
index 95003b211fd..4e69f3cd433 100644
--- a/spec/frontend/groups/components/invite_members_banner_spec.js
+++ b/spec/frontend/groups/components/invite_members_banner_spec.js
@@ -1,5 +1,5 @@
 import { shallowMount } from '@vue/test-utils';
-import { GlBanner } from '@gitlab/ui';
+import { GlBanner, GlButton } from '@gitlab/ui';
 import { mockTracking, unmockTracking } from 'helpers/tracking_helper';
 import { setCookie, parseBoolean } from '~/lib/utils/common_utils';
 import InviteMembersBanner from '~/groups/components/invite_members_banner.vue';
@@ -107,14 +107,12 @@ describe('InviteMembersBanner', () => {
   });
 
   describe('dismissing', () => {
-    const findButton = () => {
-      return wrapper.find('button');
-    };
+    const findButton = () => wrapper.findAll(GlButton).at(1);
 
     beforeEach(() => {
       wrapper = createComponent({ GlBanner });
 
-      findButton().trigger('click');
+      findButton().vm.$emit('click');
     });
 
     it('sets iDismissed to true', () => {
@@ -138,7 +136,7 @@ describe('InviteMembersBanner', () => {
     });
 
     it('does not render the banner', () => {
-      expect(wrapper.contains(GlBanner)).toBe(false);
+      expect(wrapper.find(GlBanner).exists()).toBe(false);
     });
   });
 });