diff --git a/components/_util/__tests__/wave.test.tsx b/components/_util/__tests__/wave.test.tsx
index 67fdd16ed1..e166cb2347 100644
--- a/components/_util/__tests__/wave.test.tsx
+++ b/components/_util/__tests__/wave.test.tsx
@@ -3,6 +3,7 @@ import mountTest from '../../../tests/shared/mountTest';
 import { render, waitFakeTimer, fireEvent, act } from '../../../tests/utils';
 import ConfigProvider from '../../config-provider';
 import Wave from '../wave';
+import type { InternalWave } from '../wave';
 
 describe('Wave component', () => {
   mountTest(Wave);
@@ -194,7 +195,7 @@ describe('Wave component', () => {
   });
 
   it('bindAnimationEvent should return when node is null', () => {
-    const ref = React.createRef<Wave>();
+    const ref = React.createRef<InternalWave>();
     render(
       <Wave ref={ref}>
         <button type="button" disabled>
@@ -206,7 +207,7 @@ describe('Wave component', () => {
   });
 
   it('bindAnimationEvent.onClick should return when children is hidden', () => {
-    const ref = React.createRef<Wave>();
+    const ref = React.createRef<InternalWave>();
     render(
       <Wave ref={ref}>
         <button type="button" style={{ display: 'none' }}>
@@ -218,7 +219,7 @@ describe('Wave component', () => {
   });
 
   it('bindAnimationEvent.onClick should return when children is input', () => {
-    const ref = React.createRef<Wave>();
+    const ref = React.createRef<InternalWave>();
     render(
       <Wave ref={ref}>
         <input />
diff --git a/components/_util/wave/index.tsx b/components/_util/wave/index.tsx
index 9c315de27d..c01745fbd4 100644
--- a/components/_util/wave/index.tsx
+++ b/components/_util/wave/index.tsx
@@ -70,7 +70,7 @@ export interface WaveProps {
   children?: React.ReactNode;
 }
 
-class InternalWave extends React.Component<WaveProps> {
+export class InternalWave extends React.Component<WaveProps> {
   static contextType = ConfigContext;
 
   private instance?: {