1 files changed, 5 insertions, 3 deletions
diff --git a/src/profiling/ConnectionAcknowledgedCommandHandler.cpp b/src/profiling/ConnectionAcknowledgedCommandHandler.cpp
index 0f83a3181b..f90b601b7e 100644
--- a/src/profiling/ConnectionAcknowledgedCommandHandler.cpp
+++ b/src/profiling/ConnectionAcknowledgedCommandHandler.cpp
@@ -17,10 +17,12 @@ void ConnectionAcknowledgedCommandHandler::operator()(const Packet& packet)
 {
     if (!(packet.GetPacketFamily() == 0u && packet.GetPacketId() == 1u))
     {
-        throw armnn::Exception(std::string("Expected Packet family = 0, id = 1 but received family =")
-                               + std::to_string(packet.GetPacketFamily())
-                               +" id = " + std::to_string(packet.GetPacketId()));
+        throw armnn::InvalidArgumentException(std::string("Expected Packet family = 0, id = 1 but received family = ")
+                                              + std::to_string(packet.GetPacketFamily())
+                                              + " id = " + std::to_string(packet.GetPacketId()));
     }
+
+    // Once a Connection Acknowledged packet has been received, move to the Active state immediately
     m_StateMachine.TransitionToState(ProfilingState::Active);
 }