1 files changed, 8 insertions, 5 deletions
diff --git a/src/backends/cl/test/ClWorkloadFactoryHelper.hpp b/src/backends/cl/test/ClWorkloadFactoryHelper.hpp
index 7b60b8ad15..777bc84b8a 100644
--- a/src/backends/cl/test/ClWorkloadFactoryHelper.hpp
+++ b/src/backends/cl/test/ClWorkloadFactoryHelper.hpp
@@ -9,10 +9,9 @@
 #include <backendsCommon/IMemoryManager.hpp>
 #include <backendsCommon/test/WorkloadFactoryHelper.hpp>
 
+#include <cl/ClBackend.hpp>
 #include <cl/ClWorkloadFactory.hpp>
 
-#include <arm_compute/runtime/CL/CLBufferAllocator.h>
-
 #include <boost/polymorphic_pointer_cast.hpp>
 
 namespace
@@ -21,11 +20,15 @@ namespace
 template<>
 struct WorkloadFactoryHelper<armnn::ClWorkloadFactory>
 {
-    static armnn::ClWorkloadFactory GetFactory()
+    static armnn::IBackendInternal::IMemoryManagerSharedPtr GetMemoryManager()
     {
-        armnn::IBackendInternal::IMemoryManagerSharedPtr memoryManager =
-            std::make_shared<armnn::ClMemoryManager>(std::make_unique<arm_compute::CLBufferAllocator>());
+        armnn::ClBackend backend;
+        return backend.CreateMemoryManager();
+    }
 
+    static armnn::ClWorkloadFactory GetFactory(
+        const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager)
+    {
         return armnn::ClWorkloadFactory(boost::polymorphic_pointer_downcast<armnn::ClMemoryManager>(memoryManager));
     }
 };