diff --git a/src/com/android/customization/module/ThemePickerInjector.kt b/src/com/android/customization/module/ThemePickerInjector.kt
index 98c6361..b23d912 100644
--- a/src/com/android/customization/module/ThemePickerInjector.kt
+++ b/src/com/android/customization/module/ThemePickerInjector.kt
@@ -235,7 +235,12 @@
                     repository =
                         WallpaperRepository(
                             scope = getApplicationCoroutineScope(),
-                            client = WallpaperClientImpl(context = context),
+                            client =
+                                WallpaperClientImpl(
+                                    context = context,
+                                    infoFactory = getCurrentWallpaperInfoFactory(context),
+                                    wallpaperManager = WallpaperManager.getInstance(context)
+                                ),
                             wallpaperPreferences = getPreferences(context = context),
                             backgroundDispatcher = Dispatchers.IO,
                         ),