diff --git a/packages/core/src/components/EdgeLabelRenderer/index.tsx b/packages/core/src/components/EdgeLabelRenderer/index.tsx index 059dae1b..07477493 100644 --- a/packages/core/src/components/EdgeLabelRenderer/index.tsx +++ b/packages/core/src/components/EdgeLabelRenderer/index.tsx @@ -3,12 +3,10 @@ import { createPortal } from 'react-dom'; import { useStore } from '../../hooks/useStore'; import { ReactFlowState } from '../../types'; -const selector = (s: ReactFlowState) => ({ - edgeLabelRenderer: s.domNode?.querySelector('.react-flow__edgelabel-renderer'), -}); +const selector = (s: ReactFlowState) => s.domNode?.querySelector('.react-flow__edgelabel-renderer'); function EdgeLabelRenderer({ children }: { children: ReactNode }) { - const { edgeLabelRenderer } = useStore(selector); + const edgeLabelRenderer = useStore(selector); if (!edgeLabelRenderer) { return null;