diff --git a/libraries/ui/build.gradle b/libraries/ui/build.gradle
index bdc618db59..ac81ef51c6 100644
--- a/libraries/ui/build.gradle
+++ b/libraries/ui/build.gradle
@@ -27,6 +27,11 @@ android {
             withSourcesJar()
         }
     }
+
+    sourceSets {
+        androidTest.assets.srcDir '../test_data/src/test/assets'
+        test.assets.srcDir '../test_data/src/test/assets/'
+    }
 }
 
 dependencies {
@@ -36,6 +41,8 @@ dependencies {
     implementation 'androidx.recyclerview:recyclerview:' + androidxRecyclerViewVersion
     compileOnly 'org.checkerframework:checker-qual:' + checkerframeworkVersion
     compileOnly 'org.jetbrains.kotlin:kotlin-annotations-jvm:' + kotlinAnnotationsVersion
+    testImplementation project(path: ':lib-exoplayer')
+    testImplementation project(path: ':test-utils-robolectric')
     testImplementation project(modulePrefix + 'test-utils')
     testImplementation 'org.robolectric:robolectric:' + robolectricVersion
 }