diff --git a/libs/shared/lib/data-access/store/schemaSlice.ts b/libs/shared/lib/data-access/store/schemaSlice.ts
index 76a66a7d3c755ec36e58969b0e06a26d2210b16b..785257627dc16b96e113caad00f31aad4a3f04d6 100644
--- a/libs/shared/lib/data-access/store/schemaSlice.ts
+++ b/libs/shared/lib/data-access/store/schemaSlice.ts
@@ -30,8 +30,14 @@ export const initialState: schemaSliceI = {
     edges: {},
   },
   graphStats: {
-    nodeStats: {},
-    edgeStats: {},
+    nodes: {
+      count: 0,
+      stats: {},
+    },
+    edges: {
+      count: 0,
+      stats: {},
+    },
   },
   // layoutName: 'Cytoscape_fcose',
   settings: {
diff --git a/libs/shared/lib/schema/model/FromBackend.ts b/libs/shared/lib/schema/model/FromBackend.ts
index 83df8d4ba2adec75effddddb3509f02017ba0f2f..de80178cd09943cb6772ab59443e0b06b127306d 100644
--- a/libs/shared/lib/schema/model/FromBackend.ts
+++ b/libs/shared/lib/schema/model/FromBackend.ts
@@ -52,8 +52,14 @@ export type SchemaGraphInference = {
 };
 
 export type SchemaGraphStats = {
-  nodeStats: Record<string, NodeOrEdgeStats>; // node key -> Stats
-  edgeStats: Record<string, NodeOrEdgeStats>; // edge key -> Stats
+  nodes: {
+    count: number;
+    stats: Record<string, NodeOrEdgeStats>; // node key -> Stats
+  };
+  edges: {
+    count: number;
+    stats: Record<string, NodeOrEdgeStats>; // edge key -> Stats
+  };
 };
 
 export type NodeOrEdgeStats = {