diff --git a/packages/react/src/NodeViewContent.tsx b/packages/react/src/NodeViewContent.tsx index c06bc7d7..5d867541 100644 --- a/packages/react/src/NodeViewContent.tsx +++ b/packages/react/src/NodeViewContent.tsx @@ -5,17 +5,20 @@ export interface NodeViewContentProps { as?: React.ElementType, } -export const NodeViewContent: React.FC = props => { - const Tag = props.as || 'div' +export const NodeViewContent: React.FC = + React.forwardRef((props, ref) => { + const Tag = props.as || 'div' - return ( - - ) -} + return ( + + ) + }) + diff --git a/packages/react/src/NodeViewWrapper.tsx b/packages/react/src/NodeViewWrapper.tsx index da502459..f50aa53f 100644 --- a/packages/react/src/NodeViewWrapper.tsx +++ b/packages/react/src/NodeViewWrapper.tsx @@ -6,19 +6,21 @@ export interface NodeViewWrapperProps { as?: React.ElementType, } -export const NodeViewWrapper: React.FC = props => { - const { onDragStart } = useReactNodeView() - const Tag = props.as || 'div' +export const NodeViewWrapper: React.FC = + React.forwardRef((props, ref) => { + const { onDragStart } = useReactNodeView() + const Tag = props.as || 'div' - return ( - - ) -} + return ( + + ) + })