1 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/LstmUtils.cpp b/src/backends/reference/workloads/LstmUtils.cpp
index f197aae291..8e1db0e88f 100644
--- a/src/backends/reference/workloads/LstmUtils.cpp
+++ b/src/backends/reference/workloads/LstmUtils.cpp
@@ -7,7 +7,7 @@
 
 #include "LstmUtils.hpp"
 #include "BaseIterator.hpp"
-#include <backendsCommon/CpuTensorHandle.hpp>
+#include <backendsCommon/TensorHandle.hpp>
 
 
 // Helper functions ported from the Android code base
@@ -296,12 +296,12 @@ void SetActivationParameters(uint32_t activation,
     }
 }
 
-std::unique_ptr<armnn::ScopedCpuTensorHandle> AssignScopedCpuTensorHandle(const armnn::ConstCpuTensorHandle* ptr)
+std::unique_ptr<armnn::ScopedTensorHandle> AssignScopedTensorHandle(const armnn::ConstTensorHandle *ptr)
 {
     if (!ptr)
     {
         return nullptr;
     }
 
-    return std::make_unique<armnn::ScopedCpuTensorHandle>(*ptr);
+    return std::make_unique<armnn::ScopedTensorHandle>(*ptr);
 }