diff --git a/script/test/test_pipeline2.py b/script/test/test_pipeline2.py
index 03ac12e..f13f35f 100644
--- a/script/test/test_pipeline2.py
+++ b/script/test/test_pipeline2.py
@@ -1,10 +1,17 @@
+def get_pipeline_msg():
+    if not get_pipeline():
+        raise Exception("Pipeline not started")
+    with source(host="127.0.0.1", port=pipeline_port, mode=SUB, receive_timeout = 3000) as stream:
+        data = stream.receive()
+        if not data: 
+            raise Exception("Received None message.")
+        print retuern data.data
 
 #viewer.show_stream("tcp://localhost:" + str(DEFAULT_PIPELINE_PORT))
 #pipeline_config = PipelineConfig("test_pipeline", parameters={"camera_name": "simulation", "function":"transparent"})
 #start_pipeline("test_pipeline", {"camera_name": "simulation", "function":"transparent"})
 
-with source(host="127.0.0.1", port=DEFAULT_PIPELINE_PORT, mode=SUB, receive_timeout = 3000) as stream:
-    data = stream.receive()
-    if not data: 
-        raise Exception("Received None message.")
-    print (data.data.data.keys())
+ 
+ 
+msg =  get_pipeline_msg().data
+print (msg.keys())