diff --git a/tests/unit/Android.bp b/tests/unit/Android.bp
index 249ce72..2fcd91f 100644
--- a/tests/unit/Android.bp
+++ b/tests/unit/Android.bp
@@ -31,7 +31,7 @@
         "mockito-target-minus-junit4",
         "platform-test-annotations",
         "truth-prebuilt",
-        "ub-uiautomator",
+        "androidx.test.uiautomator_uiautomator",
         // Don't add SettingsLib libraries here - you can use them directly as they are in the
         // instrumented Settings app.
     ],