diff --git a/components/icon/demo/basic.md b/components/icon/demo/basic.md
index 9b8597986e8d..d57235a7f773 100644
--- a/components/icon/demo/basic.md
+++ b/components/icon/demo/basic.md
@@ -14,7 +14,7 @@ title:
 Import icons from `infra-design-icons`, component name of icons with different theme is the icon name suffixed by the theme name. Specify the `spin` property to show spinning animation.
 
 ```jsx
-import { Space } from 'antd';
+import { Space } from 'infrad';
 import {
   HomeOutlined,
   SettingFilled,
diff --git a/components/icon/demo/two-tone.md b/components/icon/demo/two-tone.md
index 00e5c9416bad..92a7b36e0928 100644
--- a/components/icon/demo/two-tone.md
+++ b/components/icon/demo/two-tone.md
@@ -14,7 +14,7 @@ title:
 You can set `twoToneColor` prop to specific primary color for two-tone icons.
 
 ```jsx
-import { Space } from 'antd';
+import { Space } from 'infrad';
 import { SmileTwoTone, HeartTwoTone, CheckCircleTwoTone } from 'infra-design-icons';
 
 ReactDOM.render(
diff --git a/components/image/demo/controlled-preview.md b/components/image/demo/controlled-preview.md
index b9e5f3f35758..e94d3f57e7bd 100644
--- a/components/image/demo/controlled-preview.md
+++ b/components/image/demo/controlled-preview.md
@@ -15,7 +15,7 @@ You can make preview controlled.
 
 ```jsx
 import React, { useState } from 'react';
-import { Image, Button } from 'antd';
+import { Image, Button } from 'infrad';
 
 function ImageDemo() {
   const [visible, setVisible] = useState(false);