diff --git a/docs/caseShow/demos/pipelineDemo.tsx b/docs/caseShow/demos/pipelineDemo.tsx
index b9624be..b4df109 100644
--- a/docs/caseShow/demos/pipelineDemo.tsx
+++ b/docs/caseShow/demos/pipelineDemo.tsx
@@ -6,11 +6,12 @@ import {
   FlowView,
   FlowViewProvider,
   Handle,
+  NodeProps,
   Position,
+  SelectType,
   useFlowViewer,
 } from '@ant-design/pro-flow';
 import { FC, useCallback } from 'react';
-import { SelectType } from '../../../src';
 import useStyles from './index.style';
 
 interface PipeNodeChild {
@@ -32,9 +33,7 @@ interface PipeNode {
 const nodeWidth = 170;
 const nodeHeight = 500;
 
-export const PipeNode: FC<{
-  data: PipeNode;
-}> = ({ data }) => {
+export const PipeNode: FC<NodeProps<PipeNode>> = ({ data }) => {
   const { stepTitle, title, des, logo, needSwitch = false, children = [], selectType } = data;
   const { styles } = useStyles();
 
diff --git a/src/index.ts b/src/index.ts
index eaf6d29..857e122 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -31,6 +31,6 @@ export * from './Input';
 export * from './MiniMap';
 export * from './constants';
 
-export type { Connection, EdgeChange, EdgeProps, NodeChange } from 'reactflow';
+export type { Connection, EdgeChange, EdgeProps, NodeChange, NodeProps } from 'reactflow';
 export type { FlowEditorStoreProviderProps } from './FlowStoreProvider';
 export type { ExtraAction } from './NodeField';