diff --git a/include/armnnTestUtils/MockTensorHandle.hpp b/include/armnnTestUtils/MockTensorHandle.hpp
index 9a7518b..bc2f7d7 100644
--- a/include/armnnTestUtils/MockTensorHandle.hpp
+++ b/include/armnnTestUtils/MockTensorHandle.hpp
@@ -5,7 +5,13 @@
 #pragma once
 
 #include "MockMemoryManager.hpp"
+
 #include <armnn/backends/TensorHandle.hpp>
+#include <armnn/MemorySources.hpp>
+#include <armnn/Tensor.hpp>
+#include <armnn/Types.hpp>
+#include <armnn/backends/ITensorHandle.hpp>
+#include <memory>
 
 namespace armnn
 {