diff --git a/components/tag/__tests__/index.test.tsx b/components/tag/__tests__/index.test.tsx
index f40b6bad08..006ee7f4f4 100644
--- a/components/tag/__tests__/index.test.tsx
+++ b/components/tag/__tests__/index.test.tsx
@@ -110,33 +110,37 @@ describe('Tag', () => {
 
   describe('visibility', () => {
     it('can be controlled by visible with visible as initial value', () => {
-      const { asFragment, rerender } = render(<Tag visible />);
-      expect(asFragment().firstChild).toMatchSnapshot();
+      const { container, rerender } = render(<Tag visible />);
+      expect(container.querySelector('.ant-tag-hidden')).toBeFalsy();
+
       rerender(<Tag visible={false} />);
       act(() => {
         jest.runAllTimers();
       });
-      expect(asFragment().firstChild).toMatchSnapshot();
+      expect(container.querySelector('.ant-tag-hidden')).toBeTruthy();
+
       rerender(<Tag visible />);
       act(() => {
         jest.runAllTimers();
       });
-      expect(asFragment().firstChild).toMatchSnapshot();
+      expect(container.querySelector('.ant-tag-hidden')).toBeFalsy();
     });
 
     it('can be controlled by visible with hidden as initial value', () => {
-      const { asFragment, rerender } = render(<Tag visible={false} />);
-      expect(asFragment().firstChild).toMatchSnapshot();
+      const { container, rerender } = render(<Tag visible={false} />);
+      expect(container.querySelector('.ant-tag-hidden')).toBeTruthy();
+
       rerender(<Tag visible />);
       act(() => {
         jest.runAllTimers();
       });
-      expect(asFragment().firstChild).toMatchSnapshot();
+      expect(container.querySelector('.ant-tag-hidden')).toBeFalsy();
+
       rerender(<Tag visible={false} />);
       act(() => {
         jest.runAllTimers();
       });
-      expect(asFragment().firstChild).toMatchSnapshot();
+      expect(container.querySelector('.ant-tag-hidden')).toBeTruthy();
     });
   });