2 files changed, 8 insertions, 1 deletions
diff --git a/src/stateless/messaging/trex_stateless_messaging.cpp b/src/stateless/messaging/trex_stateless_messaging.cpp
index 3c6a5933..3e754649 100644
--- a/src/stateless/messaging/trex_stateless_messaging.cpp
+++ b/src/stateless/messaging/trex_stateless_messaging.cpp
@@ -47,7 +47,7 @@ TrexStatelessDpStart::handle(TrexStatelessDpCore *dp_core) {
  ************************/ 
 bool
 TrexStatelessDpStop::handle(TrexStatelessDpCore *dp_core) {
-    dp_core->stop_traffic();
+    dp_core->stop_traffic(m_port_id);
     return true;
 }
 
diff --git a/src/stateless/messaging/trex_stateless_messaging.h b/src/stateless/messaging/trex_stateless_messaging.h
index 13f6c05a..381e146d 100644
--- a/src/stateless/messaging/trex_stateless_messaging.h
+++ b/src/stateless/messaging/trex_stateless_messaging.h
@@ -72,7 +72,14 @@ private:
  */
 class TrexStatelessDpStop : public TrexStatelessCpToDpMsgBase {
 public:
+
+    TrexStatelessDpStop(uint8_t port_id) : m_port_id(port_id) {
+    }
+
     virtual bool handle(TrexStatelessDpCore *dp_core);
+
+private:
+    uint8_t m_port_id;
 };