diff --git a/tests/Compatibility/src/com/android/compatibilitytest/AppCompatibility.java b/tests/Compatibility/src/com/android/compatibilitytest/AppCompatibility.java
index b54f9be3461da..1f9e63644bde8 100644
--- a/tests/Compatibility/src/com/android/compatibilitytest/AppCompatibility.java
+++ b/tests/Compatibility/src/com/android/compatibilitytest/AppCompatibility.java
@@ -208,7 +208,8 @@ public class AppCompatibility extends InstrumentationTestCase {
     private boolean ensureForegroundActivity(RunningAppProcessInfo info) {
         Log.d(TAG, String.format("ensureForegroundActivity: proc=%s, pid=%d, state=%d",
                 info.processName, info.pid, info.processState));
-        return info.processState == ActivityManager.PROCESS_STATE_TOP;
+        return info.processState == ActivityManager.PROCESS_STATE_TOP
+            || info.processState == ActivityManager.PROCESS_STATE_TOP_SLEEPING;
     }
 
     /**