diff --git a/package.json b/package.json
index 34ffede9..f6b8de21 100644
--- a/package.json
+++ b/package.json
@@ -1,7 +1,7 @@
 {
     "name": "@d-id/client-sdk",
     "private": false,
-    "version": "1.1.55",
+    "version": "1.1.56",
     "type": "module",
     "description": "d-id client sdk",
     "repository": {
diff --git a/src/services/agent-manager/index.ts b/src/services/agent-manager/index.ts
index 7fdd0fe0..c8f43615 100644
--- a/src/services/agent-manager/index.ts
+++ b/src/services/agent-manager/index.ts
@@ -90,9 +90,10 @@ export async function createAgentManager(agent: string, options: AgentManagerOpt
     const isStreamsV2 = isStreamsV2Agent(agentEntity.presenter.type);
     analytics.enrich(getAgentInfo(agentEntity));
 
-    const { onMessage, clearQueue } = createMessageEventQueue(analytics, items, options, agentEntity, () =>
-        items.socketManager?.disconnect()
-    );
+    const { onMessage, clearQueue } = createMessageEventQueue(analytics, items, options, agentEntity, () => {
+        items.socketManager?.disconnect();
+        options.callbacks.onConnectionStateChange?.(ConnectionState.Disconnected);
+    });
 
     items.messages = getInitialMessages(options.initialMessages);