1 files changed, 7 insertions, 7 deletions
diff --git a/packages/meshbay-node/tests/test_disk_io_off_loop.py b/packages/meshbay-node/tests/test_disk_io_off_loop.py
index b77200a..e404b47 100644
--- a/packages/meshbay-node/tests/test_disk_io_off_loop.py
+++ b/packages/meshbay-node/tests/test_disk_io_off_loop.py
@@ -35,7 +35,7 @@ from meshbay_node.indexer.indexer import DirectoryIndexer
 from meshbay_node.roots import Root, RootSet
 from meshbay_node.transfers import LeaselessReads
 from meshbay_node.transport import webrtc_server
-from meshbay_node.transport.webrtc import media_tools
+from meshbay_node.transport.webrtc import files, media_tools
 from meshbay_node.transport.webrtc_server import WebRTCPeerSession
 from node_source import webrtc_files
 
@@ -130,8 +130,8 @@ def _slow(fn):
 
 async def test_a_slow_chunk_read_does_not_stop_the_loop(tmp_path, monkeypatch):
     session, _, entry, gek = await _served(tmp_path)
-    monkeypatch.setattr(webrtc_server, "_read_and_encrypt",
-                        _slow(webrtc_server._read_and_encrypt))
+    monkeypatch.setattr(files, "_read_and_encrypt",
+                        _slow(files._read_and_encrypt))
 
     with _Ticker() as ticker:
         await session._do_file_request(
@@ -156,7 +156,7 @@ async def test_a_slow_stat_does_not_stop_the_loop(tmp_path, monkeypatch):
     would then find the disk already awake — the stall moved, not removed.
     """
     session, _, entry, _ = await _served(tmp_path)
-    monkeypatch.setattr(webrtc_server, "_locate", _slow(webrtc_server._locate))
+    monkeypatch.setattr(files, "_locate", _slow(files._locate))
 
     with _Ticker() as ticker:
         await session._do_file_request(
@@ -180,7 +180,7 @@ async def test_one_root_set_reads_one_chunk_at_a_time(tmp_path):
     inside = 0
     peak = 0
     guard = threading.Lock()
-    real = webrtc_server._read_and_encrypt
+    real = files._read_and_encrypt
 
     def counting(*args, **kwargs):
         nonlocal inside, peak
@@ -194,7 +194,7 @@ async def test_one_root_set_reads_one_chunk_at_a_time(tmp_path):
             with guard:
                 inside -= 1
 
-    webrtc_server._read_and_encrypt = counting
+    files._read_and_encrypt = counting
     try:
         await asyncio.gather(*[
             session._do_file_request(
@@ -202,7 +202,7 @@ async def test_one_root_set_reads_one_chunk_at_a_time(tmp_path):
             for _ in range(6)
         ])
     finally:
-        webrtc_server._read_and_encrypt = real
+        files._read_and_encrypt = real
 
     assert peak == 1, f"{peak} reads of one root set were inside the disk at once"