diff --git a/src/browser/modules/Stream/CypherFrame/VisualizationView/PropertiesPanelContent/DetailsPane.test.tsx b/src/browser/modules/Stream/CypherFrame/VisualizationView/PropertiesPanelContent/DetailsPane.test.tsx
index 25317b7a6bf..11d8ad128e3 100644
--- a/src/browser/modules/Stream/CypherFrame/VisualizationView/PropertiesPanelContent/DetailsPane.test.tsx
+++ b/src/browser/modules/Stream/CypherFrame/VisualizationView/PropertiesPanelContent/DetailsPane.test.tsx
@@ -22,7 +22,11 @@ import React from 'react'
 
 import { DETAILS_PANE_STEP_SIZE, DetailsPane } from './DetailsPane'
 import { VizItemProperty } from 'neo4j-arc/common'
-import { GraphStyleModel, VizItem } from 'neo4j-arc/graph-visualization'
+import {
+  GraphStyleModel,
+  VizItem,
+  NodeModel
+} from 'neo4j-arc/graph-visualization'
 
 describe('<DetailsPane />', () => {
   const mockGraphStyle = new GraphStyleModel()
@@ -66,7 +70,9 @@ describe('<DetailsPane />', () => {
           item: {
             id: 'abc',
             type: 'abc2',
-            propertyList
+            propertyList,
+            source: new NodeModel('1', [], {}, {}),
+            target: new NodeModel('2', [], {}, {})
           }
         }
     }
diff --git a/src/neo4j-arc/common/components/PropertiesTable/PropertiesTable.test.tsx b/src/neo4j-arc/common/components/PropertiesTable/PropertiesTable.test.tsx
index 4e08e0d0337..6ca69352c40 100644
--- a/src/neo4j-arc/common/components/PropertiesTable/PropertiesTable.test.tsx
+++ b/src/neo4j-arc/common/components/PropertiesTable/PropertiesTable.test.tsx
@@ -20,7 +20,7 @@
 import { render, screen, waitFor } from '@testing-library/react'
 import React from 'react'
 
-import { VizItem } from 'neo4j-arc/graph-visualization'
+import { VizItem, NodeModel } from 'neo4j-arc/graph-visualization'
 
 import {
   PropertiesTable,
@@ -62,7 +62,9 @@ describe('<DetailsPane />', () => {
           item: {
             id: 'abc',
             type: 'abc2',
-            propertyList
+            propertyList,
+            source: new NodeModel('1', [], {}, {}),
+            target: new NodeModel('2', [], {}, {})
           }
         }
     }
diff --git a/src/neo4j-arc/graph-visualization/GraphVisualizer/DefaultPanelContent/DefaultDetailsPane.test.tsx b/src/neo4j-arc/graph-visualization/GraphVisualizer/DefaultPanelContent/DefaultDetailsPane.test.tsx
index 29d3a31d729..8be6bf05a76 100644
--- a/src/neo4j-arc/graph-visualization/GraphVisualizer/DefaultPanelContent/DefaultDetailsPane.test.tsx
+++ b/src/neo4j-arc/graph-visualization/GraphVisualizer/DefaultPanelContent/DefaultDetailsPane.test.tsx
@@ -27,6 +27,7 @@ import {
 import { VizItemProperty } from 'neo4j-arc/common'
 import { GraphStyleModel } from '../../models/GraphStyle'
 import { VizItem } from '../../types'
+import { NodeModel } from 'neo4j-arc/graph-visualization'
 
 describe('<DetailsPane />', () => {
   const mockGraphStyle = new GraphStyleModel()
@@ -70,7 +71,9 @@ describe('<DetailsPane />', () => {
           item: {
             id: 'abc',
             type: 'abc2',
-            propertyList
+            propertyList,
+            source: new NodeModel('1', [], {}, {}),
+            target: new NodeModel('2', [], {}, {})
           }
         }
     }
diff --git a/src/neo4j-arc/graph-visualization/index.ts b/src/neo4j-arc/graph-visualization/index.ts
index df5e5ae4ac5..b683cc49442 100644
--- a/src/neo4j-arc/graph-visualization/index.ts
+++ b/src/neo4j-arc/graph-visualization/index.ts
@@ -18,6 +18,7 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+export { NodeModel } from './models/Node'
 export { GraphModel } from './models/Graph'
 export { GraphStyleModel, Selector } from './models/GraphStyle'
 
@@ -37,7 +38,7 @@ export type { OverviewPaneProps } from './GraphVisualizer/DefaultPanelContent/De
 export {
   REL_ON_CANVAS_CREATE,
   NODE_ON_CANVAS_CREATE,
-  NODE_PROP_UPDATE, 
+  NODE_PROP_UPDATE,
   NODE_LABEL_UPDATE,
   REL_TYPE_UPDATE
 } from './GraphVisualizer/Graph/GraphEventHandlerModel'