diff --git a/src/ProFlow/index.tsx b/src/ProFlow/index.tsx
index 3eaaf9d..6fe444c 100644
--- a/src/ProFlow/index.tsx
+++ b/src/ProFlow/index.tsx
@@ -1,5 +1,5 @@
 import ProFlowController from '@/ProFlowController';
-import { useMemo, type CSSProperties, type MouseEvent as ReactMouseEvent } from 'react';
+import React, { useMemo, type CSSProperties, type MouseEvent as ReactMouseEvent } from 'react';
 import ReactFlow, { Background, BackgroundVariant, Edge, Node } from 'reactflow';
 import { ProFLowEdge, ProFlowNode } from './constants';
 import { convertMappingFrom, getRenderData } from './helper';
diff --git a/src/ProFlow/styles.tsx b/src/ProFlow/styles.tsx
index 3179466..93f1a7e 100644
--- a/src/ProFlow/styles.tsx
+++ b/src/ProFlow/styles.tsx
@@ -6,6 +6,10 @@ export const useStyles = createStyles(({ css }) => ({
     width: 100%;
     height: 100%;
 
+    .react-flow__attribution {
+      display: none;
+    }
+
     .${INIT_NODE} {
       padding: 0;
       box-sizing: border-box;
diff --git a/src/ProFlowController/index.tsx b/src/ProFlowController/index.tsx
index 05a0a69..b42bae1 100644
--- a/src/ProFlowController/index.tsx
+++ b/src/ProFlowController/index.tsx
@@ -1,6 +1,7 @@
 import { ExpandOutlined, MinusOutlined, PlusOutlined } from '@ant-design/icons';
 import { Button, Space, Tooltip } from 'antd';
 import { createStyles } from 'antd-style';
+import React from 'react';
 import { MiniMap, useReactFlow, useViewport } from 'reactflow';
 
 const useStyles = createStyles(({ css }) => ({