diff --git a/nat-lab/tests/uniffi/libtelio_proxy.py b/nat-lab/tests/uniffi/libtelio_proxy.py index dfccad918..dcf122aff 100644 --- a/nat-lab/tests/uniffi/libtelio_proxy.py +++ b/nat-lab/tests/uniffi/libtelio_proxy.py @@ -56,7 +56,7 @@ def shutdown(self, container_or_vm_name: Optional[str] = None): "Unknown" if container_or_vm_name is None else container_or_vm_name, ) - except Pyro5.errors.ConnectionClosedError as e: + except (Pyro5.errors.ConnectionClosedError, Pyro5.errors.ConnectionRefusedError) as e: # Shutting down the server via client request is naturally racy, # as sending of response is racing against process shutdown (and # thus server-side socket being closed).