diff --git a/websocket/src/websocket.cpp b/websocket/src/websocket.cpp
index a5ac74e..30215a2 100644
--- a/websocket/src/websocket.cpp
+++ b/websocket/src/websocket.cpp
@@ -504,13 +504,11 @@ static dmExtension::Result OnUpdate(dmExtension::Params* params)
             if (RESULT_OK != conn->m_Status)
             {
                 HandleCallback(conn, EVENT_ERROR, 0, conn->m_BufferSize);
-                HandleCallback(conn, EVENT_DISCONNECTED, 0, 0);
-            }
-            else
-            {
-                HandleCallback(conn, EVENT_DISCONNECTED, 0, conn->m_BufferSize);
+                conn->m_BufferSize = 0;
             }
 
+            HandleCallback(conn, EVENT_DISCONNECTED, 0, conn->m_BufferSize);
+
             g_Websocket.m_Connections.EraseSwap(i);
             --i;
             --size;