1 files changed, 7 insertions, 7 deletions
diff --git a/src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp b/src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp
index 2999be9240..116a5f14cb 100644
--- a/src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp
+++ b/src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp
@@ -5,7 +5,7 @@
 
 #include "RefAdditionUint8Workload.hpp"
 
-#include "Addition.hpp"
+#include "ArithmeticFunction.hpp"
 #include "RefWorkloadUtils.hpp"
 
 #include "Profiling.hpp"
@@ -28,12 +28,12 @@ void RefAdditionUint8Workload::Execute() const
 
     std::vector<float> results(outputInfo.GetNumElements());
 
-    Addition(inputInfo0.GetShape(),
-             inputInfo1.GetShape(),
-             outputInfo.GetShape(),
-             dequant0.data(),
-             dequant1.data(),
-             results.data());
+    ArithmeticFunction<std::plus<float>>(inputInfo0.GetShape(),
+                                         inputInfo1.GetShape(),
+                                         outputInfo.GetShape(),
+                                         dequant0.data(),
+                                         dequant1.data(),
+                                         results.data());
 
     Quantize(GetOutputTensorDataU8(0, m_Data), results.data(), outputInfo);
 }