Index: chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java |
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java |
index 26d6ff2fcf2ab2031cfae007977dcd6816a27d0b..cebba5892946d093aad447f6d67df930d02cebc4 100644 |
--- a/chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java |
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java |
@@ -70,7 +70,7 @@ public class PartnerDisableIncognitoModeIntegrationTest extends |
private void waitForParentalControlsEnabledState(final boolean parentalControlsEnabled) |
throws InterruptedException { |
- CriteriaHelper.pollForUIThreadCriteria(new Criteria() { |
+ CriteriaHelper.pollUiThread(new Criteria() { |
@Override |
public boolean isSatisfied() { |
// areParentalControlsEnabled is updated on a background thread, so we |
@@ -162,7 +162,7 @@ public class PartnerDisableIncognitoModeIntegrationTest extends |
toggleActivityForegroundState(); |
waitForParentalControlsEnabledState(true); |
- CriteriaHelper.pollForCriteria(Criteria.equals(0, new Callable<Integer>() { |
+ CriteriaHelper.pollInstrumentationThread(Criteria.equals(0, new Callable<Integer>() { |
@Override |
public Integer call() { |
return incognitoTabsCount(); |