1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/cl/workloads/ClReduceWorkload.hpp b/src/backends/cl/workloads/ClReduceWorkload.hpp
index a9c1409dcf..c63c3137bd 100644
--- a/src/backends/cl/workloads/ClReduceWorkload.hpp
+++ b/src/backends/cl/workloads/ClReduceWorkload.hpp
@@ -5,7 +5,7 @@
 
 #pragma once
 
-#include <armnn/backends/Workload.hpp>
+#include "ClBaseWorkload.hpp"
 
 #include <arm_compute/runtime/CL/functions/CLReductionOperation.h>
 
@@ -16,7 +16,7 @@ arm_compute::Status ClReduceWorkloadValidate(const TensorInfo& input,
                                              const TensorInfo& output,
                                              const ReduceDescriptor& descriptor);
 
-class ClReduceWorkload : public BaseWorkload<ReduceQueueDescriptor>
+class ClReduceWorkload : public ClBaseWorkload<ReduceQueueDescriptor>
 {
 public:
     ClReduceWorkload(const ReduceQueueDescriptor& descriptor, const WorkloadInfo& info);