diff --git a/tests/module/Android.bp b/tests/module/Android.bp
index 29d4965..f865832 100644
--- a/tests/module/Android.bp
+++ b/tests/module/Android.bp
@@ -20,7 +20,7 @@
 android_library {
     name: "ThemePickerTestModule",
     defaults: ["ThemePicker_defaults_with_overrides"],
-    srcs: ["src/com/android/customization/TestModule.kt"],
+    srcs: ["src/com/android/wallpaper/ThemePickerTestModule.kt"],
     static_libs: [
         "ThemePickerTestLib",
         "hilt_android",