diff --git a/script/test/test_stream_out.py b/script/test/test_stream_out.py
index 9897e65..8b4f647 100644
--- a/script/test/test_stream_out.py
+++ b/script/test/test_stream_out.py
@@ -29,7 +29,7 @@ def send():
 thread = Thread(target=send)
 thread.start()
 
-with source(host="127.0.0.1", port=OUTPUT_PORT, mode=SUB, receive_timeout = 3000) as stream:
+with bs_source(host="127.0.0.1", port=OUTPUT_PORT, mode=SUB, receive_timeout = 3000) as stream:
     data = stream.receive()
     if not data: 
         raise Exception("Received None message.")