diff --git a/include/armnn/backends/Workload.hpp b/include/armnn/backends/Workload.hpp
index be6fbd9..9f7aad0 100644
--- a/include/armnn/backends/Workload.hpp
+++ b/include/armnn/backends/Workload.hpp
@@ -1,5 +1,5 @@
 //
-// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 #pragma once
@@ -124,6 +124,8 @@
 
             if (!info.m_InputTensorInfos.empty())
             {
+                expectedInputType = info.m_InputTensorInfos.front().GetDataType();
+
                 if (expectedOutputType != expectedInputType)
                 {
                     ARMNN_ASSERT_MSG(false, "Trying to create workload with incorrect type");