1 files changed, 4 insertions, 1 deletions
diff --git a/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_client.py b/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_client.py
index 39d69178..6ff007b1 100644
--- a/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_client.py
+++ b/scripts/automation/trex_control_plane/stl/trex_stl_lib/trex_stl_client.py
@@ -1183,7 +1183,7 @@ class STLClient(object):
             ports : list
                 ports to execute the command
             streams: list
-                streams to attach
+                streams to attach (or profile)
 
         :returns:
             list of stream IDs in order of the stream list
@@ -1198,6 +1198,9 @@ class STLClient(object):
         ports = ports if ports is not None else self.get_acquired_ports()
         ports = self._validate_port_list(ports)
 
+        if isinstance(streams, STLProfile):
+            streams = streams.get_streams()
+
         # transform single stream
         if not isinstance(streams, list):
             streams = [streams]