1 files changed, 4 insertions, 2 deletions
diff --git a/spec/frontend/clusters/agents/components/token_table_spec.js b/spec/frontend/clusters/agents/components/token_table_spec.js
index f6baaf87fa4..6caeaf5c192 100644
--- a/spec/frontend/clusters/agents/components/token_table_spec.js
+++ b/spec/frontend/clusters/agents/components/token_table_spec.js
@@ -2,6 +2,7 @@ import { GlEmptyState, GlTooltip, GlTruncate } from '@gitlab/ui';
 import { mount } from '@vue/test-utils';
 import TokenTable from '~/clusters/agents/components/token_table.vue';
 import CreateTokenButton from '~/clusters/agents/components/create_token_button.vue';
+import CreateTokenModal from '~/clusters/agents/components/create_token_modal.vue';
 import { useFakeDate } from 'helpers/fake_date';
 import { extendedWrapper } from 'helpers/vue_test_utils_helper';
 import { MAX_LIST_COUNT } from '~/clusters/agents/constants';
@@ -50,6 +51,7 @@ describe('ClusterAgentTokenTable', () => {
 
   const findEmptyState = () => wrapper.findComponent(GlEmptyState);
   const findCreateTokenBtn = () => wrapper.findComponent(CreateTokenButton);
+  const findCreateModal = () => wrapper.findComponent(CreateTokenModal);
 
   beforeEach(() => {
     return createComponent(defaultTokens);
@@ -63,8 +65,8 @@ describe('ClusterAgentTokenTable', () => {
     expect(findCreateTokenBtn().exists()).toBe(true);
   });
 
-  it('passes the correct params to the create token component', () => {
-    expect(findCreateTokenBtn().props()).toMatchObject({
+  it('passes the correct params to the create token modal component', () => {
+    expect(findCreateModal().props()).toMatchObject({
       clusterAgentId,
       cursor,
     });