1 files changed, 2 insertions, 4 deletions
diff --git a/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java b/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
index a8d06ac33..2ab308a22 100644
--- a/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
@@ -765,9 +765,8 @@ public class WifiStateMachineTest {
         dhcpResults.addDns("8.8.8.8");
         dhcpResults.setLeaseDuration(3600);
 
-        mLooper.startAutoDispatch();
         mTestIpManager.injectDhcpSuccess(dhcpResults);
-        mLooper.stopAutoDispatch();
+        mLooper.dispatchAll();
 
         assertEquals("ConnectedState", getCurrentState().getName());
     }
@@ -794,9 +793,8 @@ public class WifiStateMachineTest {
 
         assertEquals("ObtainingIpState", getCurrentState().getName());
 
-        mLooper.startAutoDispatch();
         mTestIpManager.injectDhcpFailure();
-        mLooper.stopAutoDispatch();
+        mLooper.dispatchAll();
 
         assertEquals("DisconnectingState", getCurrentState().getName());
     }