diff --git a/mParticle-Apptentive/MPKitApptentive.m b/mParticle-Apptentive/MPKitApptentive.m
index a5df45e..8d06f4e 100644
--- a/mParticle-Apptentive/MPKitApptentive.m
+++ b/mParticle-Apptentive/MPKitApptentive.m
@@ -188,11 +188,7 @@ - (MPKitExecStatus *)receivedUserNotification:(NSDictionary *)userInfo {
 }
 
 - (MPKitExecStatus *)setDeviceToken:(NSData *)deviceToken {
-    // TODO: use `setRemoteNotificationToken` after next ApptentiveKit release (missing @objc annotation).
-#pragma clang diagnostic push
-#pragma clang diagnostic ignored "-Wdeprecated-declarations"
-    [Apptentive.shared setPushProvider:ApptentivePushProviderApptentive deviceToken:deviceToken];
-#pragma clang diagnostic pop
+    [Apptentive.shared setRemoteNotificationToken:deviceToken];
 
     return [self execStatus:MPKitReturnCodeSuccess];
 }