1 files changed, 4 insertions, 1 deletions
diff --git a/tools/test b/tools/test
index 09c2c81..69d8e7e 100755
--- a/tools/test
+++ b/tools/test
@@ -5,7 +5,9 @@ UPPER=`echo $TEST | tr 'a-z.' 'A-Z_'`
 OUTPUT=test_$TEST
 DEFINE=TEST_$UPPER
 
-COMPILE="g++ -g -Wall -Werror -D$DEFINE -o $OUTPUT $*"
+SCRIPTDIR=`dirname $0`
+
+COMPILE="g++ -I$SCRIPTDIR -g -Wall -Werror -D$DEFINE -o $OUTPUT $*"
 
 echo -e "\033[0;2mTesting $TEST:"
 echo Testing $TEST: > $OUTPUT.log
@@ -26,6 +28,7 @@ if ${COMPILE} >> ${OUTPUT}.log 2>&1; then
 	else
 		echo -e "\r\t\t\t\t\t\t[\033[1;31mFailure\033[0;2m]"
 		echo "[Failure]" >> $OUTPUT.log
+		rm -f $OUTPUT
 	fi
 else
 	echo -e "\r\t\t\t\t\t\t[\033[1;31mFailure\033[0;2m]"