diff --git a/test/framework/test_log.h b/test/framework/test_log.h
index 149afe2..801d538 100644
--- a/test/framework/test_log.h
+++ b/test/framework/test_log.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021, Arm Limited. All rights reserved.
+ * Copyright (c) 2021-2022, Arm Limited. All rights reserved.
  *
  * SPDX-License-Identifier: BSD-3-Clause
  *
@@ -8,7 +8,7 @@
 #ifndef __TEST_LOG_H__
 #define __TEST_LOG_H__
 
-#if DOMAIN_NS == 1
+#if (DOMAIN_NS == 1) || defined(TEST_BL2)
 #include "tfm_log_raw.h"
 #else
 #include "tfm_sp_log.h"
@@ -18,7 +18,7 @@
 extern "C" {
 #endif
 
-#if DOMAIN_NS == 1
+#if (DOMAIN_NS == 1) || defined(TEST_BL2)
 #define TEST_LOG(...) tfm_log_printf(__VA_ARGS__)
 #else
 #define TEST_LOG(...) tfm_sp_log_printf(__VA_ARGS__)