diff --git a/src/backends/cl/workloads/ClChannelShuffleWorkload.hpp b/src/backends/cl/workloads/ClChannelShuffleWorkload.hpp
index 972b2dd..9002e66 100644
--- a/src/backends/cl/workloads/ClChannelShuffleWorkload.hpp
+++ b/src/backends/cl/workloads/ClChannelShuffleWorkload.hpp
@@ -5,7 +5,7 @@
 
 #pragma once
 
-#include <armnn/backends/Workload.hpp>
+#include "ClBaseWorkload.hpp"
 
 #include <arm_compute/core/Error.h>
 #include <arm_compute/runtime/CL/functions/CLChannelShuffleLayer.h>
@@ -17,7 +17,7 @@
                                              const TensorInfo& output,
                                              const ChannelShuffleDescriptor& descriptor);
 
-class ClChannelShuffleWorkload : public BaseWorkload<ChannelShuffleQueueDescriptor>
+class ClChannelShuffleWorkload : public ClBaseWorkload<ChannelShuffleQueueDescriptor>
 {
 public:
     ClChannelShuffleWorkload(const ChannelShuffleQueueDescriptor& descriptor,