diff --git a/Android.mk b/Android.mk
index 987c875..64f452a 100644
--- a/Android.mk
+++ b/Android.mk
@@ -26,7 +26,7 @@
 LOCAL_MODULE_TAGS := optional
 
 LOCAL_STATIC_ANDROID_LIBRARIES := WallpaperPicker2CommonDepsLib
-LOCAL_STATIC_JAVA_LIBRARIES := styleprotosnano SystemUI-statsd SystemUISharedLib
+LOCAL_STATIC_JAVA_LIBRARIES := styleprotoslite SystemUI-statsd SystemUISharedLib
 LOCAL_PROGUARD_ENABLED := disabled
 
 LOCAL_PACKAGE_NAME := ThemePicker
diff --git a/src/com/android/customization/module/StatsLogUserEventLogger.java b/src/com/android/customization/module/StatsLogUserEventLogger.java
index b6405d5..2fc1f09 100644
--- a/src/com/android/customization/module/StatsLogUserEventLogger.java
+++ b/src/com/android/customization/module/StatsLogUserEventLogger.java
@@ -37,7 +37,7 @@
 
 import android.app.WallpaperManager;
 import android.content.Intent;
-import android.stats.style.nano.StyleEnums;
+import android.stats.style.StyleEnums;
 
 import androidx.annotation.Nullable;