1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/layers/DebugLayer.hpp b/src/armnn/layers/DebugLayer.hpp
index 39ef9c7ae0..6aaa271807 100644
--- a/src/armnn/layers/DebugLayer.hpp
+++ b/src/armnn/layers/DebugLayer.hpp
@@ -4,12 +4,12 @@
 //
 #pragma once
 
-#include <Layer.hpp>
+#include "LayerWithParameters.hpp"
 
 namespace armnn
 {
 
-class DebugLayer : public Layer
+class DebugLayer : public LayerWithParameters<DebugDescriptor>
 {
 public:
     virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph,
@@ -20,7 +20,7 @@ public:
     void ValidateTensorShapesFromInputs() override;
 
 protected:
-    DebugLayer(const char* name);
+    DebugLayer(const DebugDescriptor& param, const char* name);
     ~DebugLayer() = default;
 };