diff --git a/tests/tests.sh b/tests/tests.sh
index 938607e..fa8ffcd 100644
--- a/tests/tests.sh
+++ b/tests/tests.sh
@@ -41,7 +41,9 @@
 			grep -qE '^'$key'$' tests/knownKeys && continue
 			#Run the ag only on phh's machine. Assume that knownKeys is full enough.
 			#If it's enough, ask phh to update it
-			if [ -d /build/AOSP-9.0 ] && ag '"'$key'"' /build/AOSP-9.0/frameworks/base/core/res/res > /dev/null;then
+			if [ -d /build/AOSP-9.0 ] && \
+				(ag '"'$key'"' /build/AOSP-9.0/frameworks/base/core/res/res || \
+				ag '"'$key'"' /build/AOSP-8.1/frameworks/base/core/res/res)> /dev/null ;then
 				echo $key >> tests/knownKeys
 			else
 				echo $xml defines a non-existing attribute $key