1 files changed, 3 insertions, 2 deletions
diff --git a/packages/meshbay-node/tests/test_stream_capacity_config.py b/packages/meshbay-node/tests/test_stream_capacity_config.py
index 7c33a2f..cf90e22 100644
--- a/packages/meshbay-node/tests/test_stream_capacity_config.py
+++ b/packages/meshbay-node/tests/test_stream_capacity_config.py
@@ -25,6 +25,7 @@ from pathlib import Path
 import pytest
 
 from meshbay_node.config import load_config
+from meshbay_node.roots import RootSet
 from meshbay_node.transport.webrtc_server import (
     MAX_CONCURRENT_TRANSCODES,
     WebRTCPeerSession,
@@ -95,7 +96,7 @@ class _FakePC:
 def _semaphore_size(n):
     t = WebRTCTransport(
         sk_node=None, hub_pk_pem=b"", gek=b"\0" * 32,
-        shared_root=Path("/tmp"), index=None, max_concurrent_streams=n)
+        roots=RootSet(), index=None, max_concurrent_streams=n)
     s = WebRTCPeerSession(_FakePC(), t._ctx, peer_id="p")
     return s._transcode_semaphore()._value
 
@@ -117,7 +118,7 @@ def test_the_budget_is_shared_between_peers():
     """
     t = WebRTCTransport(
         sk_node=None, hub_pk_pem=b"", gek=b"\0" * 32,
-        shared_root=Path("/tmp"), index=None, max_concurrent_streams=2)
+        roots=RootSet(), index=None, max_concurrent_streams=2)
 
     async def go():
         a = WebRTCPeerSession(_FakePC(), t._ctx, peer_id="a")