1 files changed, 20 insertions, 2 deletions
diff --git a/src/backends/reference/test/RefTensorHandleTests.cpp b/src/backends/reference/test/RefTensorHandleTests.cpp
index 3504f53bc7..6f608e8541 100644
--- a/src/backends/reference/test/RefTensorHandleTests.cpp
+++ b/src/backends/reference/test/RefTensorHandleTests.cpp
@@ -1,11 +1,29 @@
 //
-// Copyright © 2017 Arm Ltd. All rights reserved.
+// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
+
+
+#include <doctest/doctest.h>
+#include <armnn/BackendId.hpp>
+#include <armnn/INetwork.hpp>
+#include <armnn/Tensor.hpp>
+#include <armnn/Types.hpp>
+#include <armnn/backends/ITensorHandle.hpp>
+#include <armnn/backends/ITensorHandleFactory.hpp>
+#include <armnn/backends/TensorHandle.hpp>
+#include <armnn/utility/Assert.hpp>
 #include <reference/RefTensorHandle.hpp>
 #include <reference/RefTensorHandleFactory.hpp>
+#include <reference/RefMemoryManager.hpp>
+#include <memory>
+#include <vector>
 
-#include <doctest/doctest.h>
+namespace armnn 
+{
+class Exception;
+class NullPointerException;
+}
 
 TEST_SUITE("RefTensorHandleTests")
 {