diff --git a/tests/unit/props.spec.ts b/tests/unit/props.spec.ts
index cddc781..654a861 100644
--- a/tests/unit/props.spec.ts
+++ b/tests/unit/props.spec.ts
@@ -57,8 +57,8 @@ describe('Expand', () => {
 describe('Checkbox', () => {
   test('showCheckbox', async () => {
     await wrapper.setProps({ showCheckbox: true });
-    expect(wrapper.find('.checkbox').exists()).toBe(true);
+    expect(wrapper.find('.xg-checkbox').exists()).toBe(true);
     await wrapper.setProps({ showCheckbox: false });
-    expect(wrapper.find('.checkbox').exists()).toBe(false);
+    expect(wrapper.find('.xg-checkbox').exists()).toBe(false);
   });
 });