Fix ping

truenas · Jul 10, 2024 · 5ab9fac · 5ab9fac
1 parent a935df2
commit 5ab9fac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/truenas_api_client/__init__.py b/truenas_api_client/__init__.py
@@ -531,8 +531,8 @@ def unsubscribe(self, id_):
                 self._event_callbacks.pop(k)
 
     def ping(self, timeout=10):
-        c = self.call('core.ping', background=True)
-        return c.wait(timeout)
+        c = self.call('core.ping', background=True, register_call=True)
+        return self.wait(c, timeout=timeout)
 
     def close(self):
         self._ws.close()