diff --git a/expect/tftf.exp b/expect/tftf.exp
index 8d5d611..3152579 100644
--- a/expect/tftf.exp
+++ b/expect/tftf.exp
@@ -3,14 +3,16 @@
 #
 # SPDX-License-Identifier: BSD-3-Clause
 #
-
-#
 # Expect script for Trusted Firmware Test Framework
 #
 
 source [file join [file dirname [info script]] handle-arguments.inc]
 
+source [file join [file dirname [info script]] trusted-firmware.inc]
+
 expect_string "Booting trusted firmware test framework" "Starting TFTF"
+expect_re "Running at NS-EL(1|2)"
+
 expect {
 	"Tests Failed  : 0" {
 		puts "<<TFTF Success>>"