diff --git a/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.h b/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.h
index 4de41a97f..7aba981e2 100644
--- a/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.h
+++ b/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.h
@@ -10,7 +10,7 @@ NS_ASSUME_NONNULL_BEGIN
 @interface OWSPrimaryStorage (PreKeyStore) <PreKeyStore>
 
 - (NSArray<PreKeyRecord *> *)generatePreKeyRecords;
-- (NSArray<PreKeyRecord *> *)generatePreKeyRecords:(int)amount;
+- (NSArray<PreKeyRecord *> *)generatePreKeyRecords:(NSUInteger)batchSize;
 - (void)storePreKeyRecords:(NSArray<PreKeyRecord *> *)preKeyRecords NS_SWIFT_NAME(storePreKeyRecords(_:));
 
 @end
diff --git a/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.m b/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.m
index b354f9b8f..13c77e5f8 100644
--- a/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.m
+++ b/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.m
@@ -18,7 +18,12 @@ NS_ASSUME_NONNULL_BEGIN
 
 @implementation OWSPrimaryStorage (PreKeyStore)
 
-- (NSArray<PreKeyRecord *> *)generatePreKeyRecords:(int)batchSize
+- (NSArray<PreKeyRecord *> *)generatePreKeyRecords
+{
+    return [self generatePreKeyRecords:BATCH_SIZE];
+}
+
+- (NSArray<PreKeyRecord *> *)generatePreKeyRecords:(NSUInteger)batchSize
 {
     NSMutableArray *preKeyRecords = [NSMutableArray array];
 
@@ -42,11 +47,6 @@ NS_ASSUME_NONNULL_BEGIN
     return preKeyRecords;
 }
 
-- (NSArray<PreKeyRecord *> *)generatePreKeyRecords
-{
-    return [self generatePreKeyRecords:BATCH_SIZE];
-}
-
 - (void)storePreKeyRecords:(NSArray<PreKeyRecord *> *)preKeyRecords
 {
     for (PreKeyRecord *record in preKeyRecords) {