diff --git a/core/embed/extmod/modtrezorcrypto/modtrezorcrypto-aes.h b/core/embed/extmod/modtrezorcrypto/modtrezorcrypto-aes.h
index 34c84cfe9..ed1f009d8 100644
--- a/core/embed/extmod/modtrezorcrypto/modtrezorcrypto-aes.h
+++ b/core/embed/extmod/modtrezorcrypto/modtrezorcrypto-aes.h
@@ -36,6 +36,11 @@ enum AESMode {
 ///     """
 ///     AES context.
 ///     """
+///     ECB: int
+///     CBC: int
+///     CFB: int
+///     OFB: int
+///     CTR: int
 typedef struct _mp_obj_AES_t {
   mp_obj_base_t base;
   aes_encrypt_ctx encrypt_ctx;
diff --git a/core/mocks/generated/trezorcrypto/__init__.pyi b/core/mocks/generated/trezorcrypto/__init__.pyi
index 446b897ab..4f29d66e0 100644
--- a/core/mocks/generated/trezorcrypto/__init__.pyi
+++ b/core/mocks/generated/trezorcrypto/__init__.pyi
@@ -6,6 +6,11 @@ class aes:
     """
     AES context.
     """
+    ECB: int
+    CBC: int
+    CFB: int
+    OFB: int
+    CTR: int
 
     def __init__(self, mode: int, key: bytes, iv: bytes = None) -> None:
         """