diff --git a/src/readers/queryService.ts b/src/readers/queryService.ts
index 4609e9363341d77ad898702ae7ee2ab4cb955347..0973c02d204a896030b376c894bd90db0d3edfa3 100644
--- a/src/readers/queryService.ts
+++ b/src/readers/queryService.ts
@@ -7,7 +7,11 @@ import { query2Cypher } from '../utils/cypher/converter';
 import { parseCountCypherQuery, parseCypherQuery } from '../utils/cypher/queryParser';
 import { formatTimeDifference } from 'ts-common/src/logger/logger';
 import { Query2BackendQuery } from '../utils/reactflow/query2backend';
-import type { GraphQueryResultFromBackend, GraphQueryResultMetaFromBackend } from 'ts-common/src/model/webSocket/graphResult';
+import type {
+  CountQueryResultFromBackend,
+  GraphQueryResultFromBackend,
+  GraphQueryResultMetaFromBackend,
+} from 'ts-common/src/model/webSocket/graphResult';
 import { RabbitMqBroker } from 'ts-common/rabbitMq';
 import { Neo4jConnection } from 'ts-common/neo4j';
 import type { QueryCypher } from '../utils/cypher/converter/queryConverter';
@@ -190,7 +194,7 @@ export const queryServiceReader = async (frontendPublisher: RabbitMqBroker, mlPu
           const result = await queryService(ss.dbConnections[i], cypher, message.useCached);
 
           // Cache nodeCounts such that we can display differentiation for each query
-          await ums.updateQuery(headers.message.sessionData.userID, message.saveStateID, {
+          await ums.updateQuery(ss.userId, message.saveStateID, {
             ...activeQueryInfo,
             graph: {
               ...activeQueryInfo.graph,