1 files changed, 6 insertions, 5 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 f98eceb..b77200a 100644
--- a/packages/meshbay-node/tests/test_disk_io_off_loop.py
+++ b/packages/meshbay-node/tests/test_disk_io_off_loop.py
@@ -35,6 +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_server import WebRTCPeerSession
 from node_source import webrtc_files
 
@@ -387,22 +388,22 @@ async def test_ffmpeg_output_over_the_cap_is_refused_before_it_is_read(tmp_path)
     scratch.write_bytes(b"x" * 5000)
 
     with pytest.raises(RuntimeError, match=r"5000 bytes, over the 1024 cap"):
-        webrtc_server._read_scratch_capped(scratch, 1024, "transcoded audio")
+        media_tools._read_scratch_capped(scratch, 1024, "transcoded audio")
 
     # And under the cap it simply reads.
-    assert webrtc_server._read_scratch_capped(scratch, 8192, "x") == b"x" * 5000
+    assert media_tools._read_scratch_capped(scratch, 8192, "x") == b"x" * 5000
 
 
 async def test_a_slow_scratch_read_does_not_stop_the_loop(tmp_path, monkeypatch):
     """Measured like the others: the loop keeps its wake-ups during the read."""
     scratch = tmp_path / "out.vtt"
     scratch.write_bytes(CONTENT)
-    monkeypatch.setattr(webrtc_server, "_read_scratch_capped",
-                        _slow(webrtc_server._read_scratch_capped))
+    monkeypatch.setattr(media_tools, "_read_scratch_capped",
+                        _slow(media_tools._read_scratch_capped))
 
     with _Ticker() as ticker:
         blob = await asyncio.to_thread(
-            webrtc_server._read_scratch_capped, scratch, 1 << 20, "subtitle track")
+            media_tools._read_scratch_capped, scratch, 1 << 20, "subtitle track")
 
     assert blob == CONTENT
     assert ticker.ticks > SLOW_S / TICK_S / 2, (