1 files changed, 4 insertions, 6 deletions
diff --git a/packages/meshbay-node/tests/test_hot_reload_survives_client_close.py b/packages/meshbay-node/tests/test_hot_reload_survives_client_close.py
index fc6af70..fb9eade 100644
--- a/packages/meshbay-node/tests/test_hot_reload_survives_client_close.py
+++ b/packages/meshbay-node/tests/test_hot_reload_survives_client_close.py
@@ -18,17 +18,15 @@ import asyncio
 import base64
 import os
 from pathlib import Path
+from unittest.mock import AsyncMock, MagicMock, patch
 
+import meshbay_node.indexer.indexer as indexer_mod
 import pytest
 from cryptography.hazmat.primitives import serialization
 from cryptography.hazmat.primitives.asymmetric.ed25519 import Ed25519PrivateKey
 from cryptography.hazmat.primitives.asymmetric.x25519 import X25519PrivateKey
-from unittest.mock import AsyncMock, MagicMock, patch
-
 from meshbay_node import ops
-from meshbay_node.config import Config, HubConfig, NodeConfig, GroupConfig, KeystoreConfig
 from meshbay_node.daemon import NodeDaemon
-import meshbay_node.indexer.indexer as indexer_mod
 
 
 def _free_port() -> int:
@@ -149,7 +147,7 @@ async def test_hot_loaded_group_finishes_scanning_without_anyone_awaiting_the_re
             with patch("signal.SIGINT", 2), patch("signal.SIGTERM", 15):
                 try:
                     await asyncio.wait_for(daemon.run(), timeout=15)
-                except (asyncio.TimeoutError, Exception):
+                except (TimeoutError, Exception):
                     pass
 
         run_task = asyncio.create_task(run_daemon())
@@ -270,7 +268,7 @@ async def test_group_scoped_ops_404_until_listed_then_succeed(tmp_path):
             with patch("signal.SIGINT", 2), patch("signal.SIGTERM", 15):
                 try:
                     await asyncio.wait_for(daemon.run(), timeout=15)
-                except (asyncio.TimeoutError, Exception):
+                except (TimeoutError, Exception):
                     pass
 
         run_task = asyncio.create_task(run_daemon())