diff --git a/src/com/android/launcher3/compat/UserManagerCompatVL.java b/src/com/android/launcher3/compat/UserManagerCompatVL.java
index da7b116..ad6f78e 100644
--- a/src/com/android/launcher3/compat/UserManagerCompatVL.java
+++ b/src/com/android/launcher3/compat/UserManagerCompatVL.java
@@ -32,6 +32,7 @@
         super(context);
     }
 
+    @Override
     public List<UserHandleCompat> getUserProfiles() {
         List<UserHandle> users = mUserManager.getUserProfiles();
         if (users == null) {
@@ -45,16 +46,17 @@
         return compatUsers;
     }
 
+    @Override
     public Drawable getBadgedDrawableForUser(Drawable unbadged, UserHandleCompat user) {
-        return mUserManager.getBadgedDrawableForUser(unbadged, user.getUser());
+        return mUserManager.getBadgedIconForUser(unbadged, user.getUser());
     }
 
+    @Override
     public CharSequence getBadgedLabelForUser(CharSequence label, UserHandleCompat user) {
         if (user == null) {
             return label;
         }
-        // TODO Remove casts when API has made it from lmp-dev -> sdk
-        return (CharSequence) mUserManager.getBadgedLabelForUser((String) label, user.getUser());
+        return mUserManager.getBadgedLabelForUser(label, user.getUser());
     }
 }