diff --git a/quickstep/tests/src/com/android/quickstep/InputConsumerUtilsTest.java b/quickstep/tests/src/com/android/quickstep/InputConsumerUtilsTest.java
index f4bfd5ee74..8a6aa5fb49 100644
--- a/quickstep/tests/src/com/android/quickstep/InputConsumerUtilsTest.java
+++ b/quickstep/tests/src/com/android/quickstep/InputConsumerUtilsTest.java
@@ -201,8 +201,10 @@ public class InputConsumerUtilsTest {
 
     @After
     public void cleanUp() {
-        mInputMonitorCompat.dispose();
-        mInputEventReceiver.dispose();
+        runOnMainSync(() -> {
+            mInputMonitorCompat.dispose();
+            mInputEventReceiver.dispose();
+        });
     }
 
     @Test