diff --git a/scripts/mbedtls_dev/bignum_common.py b/scripts/mbedtls_dev/bignum_common.py
index 81bc28e..3ff8b2f 100644
--- a/scripts/mbedtls_dev/bignum_common.py
+++ b/scripts/mbedtls_dev/bignum_common.py
@@ -251,10 +251,10 @@
         # provides earlier/more robust input validation.
         self.int_n = hex_to_int(val_n)
 
-    def to_montgomery(self, val) -> int:
+    def to_montgomery(self, val: int) -> int:
         return (val * self.r) % self.int_n
 
-    def from_montgomery(self, val) -> int:
+    def from_montgomery(self, val: int) -> int:
         return (val * self.r_inv) % self.int_n
 
     @property