1 files changed, 11 insertions, 8 deletions
diff --git a/src/profiling/RequestCounterDirectoryCommandHandler.hpp b/src/profiling/RequestCounterDirectoryCommandHandler.hpp
index a03300af48..02bf64d17a 100644
--- a/src/profiling/RequestCounterDirectoryCommandHandler.hpp
+++ b/src/profiling/RequestCounterDirectoryCommandHandler.hpp
@@ -8,6 +8,7 @@
 #include "CommandHandlerFunctor.hpp"
 #include "ISendCounterPacket.hpp"
 #include "Packet.hpp"
+#include "ProfilingStateMachine.hpp"
 
 namespace armnn
 {
@@ -19,23 +20,25 @@ class RequestCounterDirectoryCommandHandler : public CommandHandlerFunctor
 {
 
 public:
-    RequestCounterDirectoryCommandHandler(uint32_t packetId, uint32_t version,
+    RequestCounterDirectoryCommandHandler(uint32_t packetId,
+                                          uint32_t version,
                                           ICounterDirectory& counterDirectory,
-                                          ISendCounterPacket& sendCounterPacket)
-    : CommandHandlerFunctor(packetId, version),
-    m_CounterDirectory(counterDirectory),
-    m_SendCounterPacket(sendCounterPacket)
+                                          ISendCounterPacket& sendCounterPacket,
+                                          ProfilingStateMachine& profilingStateMachine)
+        : CommandHandlerFunctor(packetId, version)
+        , m_CounterDirectory(counterDirectory)
+        , m_SendCounterPacket(sendCounterPacket)
+        , m_StateMachine(profilingStateMachine)
     {}
 
     void operator()(const Packet& packet) override;
 
-
 private:
-    ICounterDirectory& m_CounterDirectory;
+    const ICounterDirectory& m_CounterDirectory;
     ISendCounterPacket& m_SendCounterPacket;
+    const ProfilingStateMachine& m_StateMachine;
 };
 
 } // namespace profiling
 
 } // namespace armnn
-