diff --git a/tests/ImageTensorGenerator/ImageTensorGenerator.cpp b/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
index 4e8fe78..9c23200 100644
--- a/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
+++ b/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
@@ -225,7 +225,7 @@
         }
         else if (m_OutputType == "qasymm8")
         {
-            return armnn::DataType::QuantisedAsymm8;
+            return armnn::DataType::QAsymmU8;
         }
         else
         {
@@ -272,7 +272,7 @@
                 imageDataContainers.push_back(PrepareImageTensor<int>(
                     imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
                 break;
-            case armnn::DataType::QuantisedAsymm8:
+            case armnn::DataType::QAsymmU8:
                 imageDataContainers.push_back(PrepareImageTensor<uint8_t>(
                     imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
                 break;
diff --git a/tests/ImageTensorGenerator/ImageTensorGenerator.hpp b/tests/ImageTensorGenerator/ImageTensorGenerator.hpp
index b1cb5e3..4793f82 100644
--- a/tests/ImageTensorGenerator/ImageTensorGenerator.hpp
+++ b/tests/ImageTensorGenerator/ImageTensorGenerator.hpp
@@ -60,7 +60,7 @@
                 case armnn::DataType::Signed32:
                     normParams.mean = { 128.0, 128.0, 128.0 };
                     break;
-                case armnn::DataType::QuantisedAsymm8:
+                case armnn::DataType::QAsymmU8:
                 default:
                     break;
             }