From 200edeefa3082d5307700ffccb70f66a0464f48a Mon Sep 17 00:00:00 2001
From: Alan Fleming <>
Date: Mon, 11 Mar 2024 10:11:38 +1100
Subject: [PATCH] Fix reject 'Control comm was closed too early' when comm
 closed as expected.

---
 packages/base-manager/src/manager-base.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/base-manager/src/manager-base.ts b/packages/base-manager/src/manager-base.ts
index 4953767894..9ccddcc12a 100644
--- a/packages/base-manager/src/manager-base.ts
+++ b/packages/base-manager/src/manager-base.ts
@@ -413,7 +413,10 @@ export abstract class ManagerBase implements IWidgetManager {
           resolve(null);
         });
 
-        initComm.on_close(() => reject('Control comm was closed too early'));
+        initComm.on_close(() => {
+          if (data.method !== 'update_states')
+            reject('Control comm was closed too early');
+        });
 
         // Send a states request msg
         initComm.send({ method: 'request_states' }, {});