diff --git a/components/notification/demo/with-btn.tsx b/components/notification/demo/with-btn.tsx
index 77af65afe5..e7a602f655 100644
--- a/components/notification/demo/with-btn.tsx
+++ b/components/notification/demo/with-btn.tsx
@@ -14,10 +14,10 @@ const App: React.FC = () => {
     const key = `open${Date.now()}`;
     const btn = (
       <Space>
-        <Button type="link" size="small" onClick={() => notification.destroy()}>
+        <Button type="link" size="small" onClick={() => api.destroy()}>
           Destroy All
         </Button>
-        <Button type="primary" size="small" onClick={() => notification.destroy(key)}>
+        <Button type="primary" size="small" onClick={() => api.destroy(key)}>
           Confirm
         </Button>
       </Space>