diff --git a/packages/core/src/types/edge.ts b/packages/core/src/types/edge.ts
index 3e92e4ce..e989f006 100644
--- a/packages/core/src/types/edge.ts
+++ b/packages/core/src/types/edge.ts
@@ -172,7 +172,7 @@ export interface EdgeProps<Data = ElementData, CustomEvents = {}, Type extends s
   type: Type
   label?: string | VNode | Component<EdgeTextProps> | Object
   style?: CSSProperties
-  selected: boolean
+  selected?: boolean
   sourcePosition: Position
   targetPosition: Position
   sourceHandleId?: string
diff --git a/packages/core/src/types/store.ts b/packages/core/src/types/store.ts
index 54baa49d..9b9fa2d8 100644
--- a/packages/core/src/types/store.ts
+++ b/packages/core/src/types/store.ts
@@ -238,7 +238,7 @@ export interface Actions extends ViewportFunctions {
   /** apply translate extent to d3 */
   setTranslateExtent: (translateExtent: CoordinateExtent) => void
   /** apply extent to nodes */
-  setNodeExtent: (nodeExtent: CoordinateExtent) => void
+  setNodeExtent: (nodeExtent: CoordinateExtent | CoordinateExtentRange) => void
   /** enable/disable node interaction (dragging, selecting etc) */
   setInteractive: (isInteractive: boolean) => void
   /** set new state */
diff --git a/packages/core/src/utils/drag.ts b/packages/core/src/utils/drag.ts
index 60bb6d7a..a603542c 100644
--- a/packages/core/src/utils/drag.ts
+++ b/packages/core/src/utils/drag.ts
@@ -103,7 +103,7 @@ function getParentExtent(
 export function getExtent<T extends NodeDragItem | GraphNode>(
   item: T,
   onError: State['hooks']['error']['trigger'],
-  extent?: CoordinateExtent,
+  extent?: State['nodeExtent'],
   parent?: GraphNode,
 ) {
   let currentExtent = item.extent || extent
@@ -146,7 +146,7 @@ export function calcNextPosition(
   node: GraphNode | NodeDragItem,
   nextPosition: XYPosition,
   onError: State['hooks']['error']['trigger'],
-  nodeExtent?: CoordinateExtent,
+  nodeExtent?: State['nodeExtent'],
   parentNode?: GraphNode,
 ) {
   const extent = getExtent(node, onError, nodeExtent, parentNode)