diff --git a/components/message/demo/hooks.tsx b/components/message/demo/hooks.tsx
index f88ded3157..9060049c2d 100644
--- a/components/message/demo/hooks.tsx
+++ b/components/message/demo/hooks.tsx
@@ -1,28 +1,20 @@
-import React, { useMemo } from 'react';
+import React from 'react';
 import { Button, message } from 'antd';
 
-const Context = React.createContext({ name: 'Default' });
-
 const App: React.FC = () => {
   const [messageApi, contextHolder] = message.useMessage();
 
   const info = () => {
-    messageApi.open({
-      type: 'info',
-      content: <Context.Consumer>{({ name }) => `Hello, ${name}!`}</Context.Consumer>,
-      duration: 1,
-    });
+    messageApi.info('Hello, Ant Design!');
   };
 
-  const contextValue = useMemo(() => ({ name: 'Ant Design' }), []);
-
   return (
-    <Context.Provider value={contextValue}>
+    <>
       {contextHolder}
       <Button type="primary" onClick={info}>
         Display normal message
       </Button>
-    </Context.Provider>
+    </>
   );
 };