diff --git a/package/ios/AdvancedTextInputViewContainer.h b/package/ios/AdvancedTextInputViewContainer.h
index 3b55fb2..a3a8539 100644
--- a/package/ios/AdvancedTextInputViewContainer.h
+++ b/package/ios/AdvancedTextInputViewContainer.h
@@ -29,7 +29,7 @@
 - (void)setDefaultValue:(NSString *)defaultValue;
 - (void)setValue:(NSString *)value;
 - (void)setAffinityFormat:(NSArray<NSString *> *)affinityFormat;
-- (void)setAffinityCalculationStrategy:(NSInteger)affinityCalculationStrategy;
+- (void)setAffinityCalculationStrategy:(NSNumber *)affinityCalculationStrategy;
 - (void)setValidationRegex:(NSString *)validationRegex;
 - (void)cleanup;
 - (void)setMaskedText:(NSString *_Nonnull)text autocomplete:(BOOL)autocomplete;
diff --git a/package/ios/AdvancedtextInputMaskDecoratorView.mm b/package/ios/AdvancedtextInputMaskDecoratorView.mm
index b22204f..95f85b3 100644
--- a/package/ios/AdvancedtextInputMaskDecoratorView.mm
+++ b/package/ios/AdvancedtextInputMaskDecoratorView.mm
@@ -121,7 +121,7 @@ - (void)updateProps:(const facebook::react::Props::Shared &)props
   }
 
   if (newViewProps.affinityCalculationStrategy != oldViewProps.affinityCalculationStrategy) {
-    [_view setAffinityCalculationStrategy:newViewProps.affinityCalculationStrategy];
+    [_view setAffinityCalculationStrategy:@(newViewProps.affinityCalculationStrategy)];
   }
 
   if (newViewProps.validationRegex != oldViewProps.validationRegex) {