diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/storage/AccountsManager.java b/service/src/main/java/org/whispersystems/textsecuregcm/storage/AccountsManager.java
index 9294f3116..c6ccd6425 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/storage/AccountsManager.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/storage/AccountsManager.java
@@ -767,7 +767,7 @@ public class AccountsManager extends RedisPubSubAdapter<String, String> implemen
     return updateWithRetries(
         account,
         a -> {
-          setPniKeys(account, pniIdentityKey, pniRegistrationIds);
+          setPniKeys(a, pniIdentityKey, pniRegistrationIds);
           return true;
         },
         a -> accounts.changeNumber(a, targetNumber, targetPhoneNumberIdentifier, maybeDisplacedUuid, keyWriteItems),