1 files changed, 6 insertions, 1 deletions
diff --git a/accounts/utils/confirmation.py b/accounts/utils/confirmation.py
index b75716d..60967de 100644
--- a/accounts/utils/confirmation.py
+++ b/accounts/utils/confirmation.py
@@ -3,6 +3,8 @@ from itsdangerous import BadSignature, SignatureExpired, URLSafeTimedSerializer
 from werkzeug.exceptions import Forbidden
 from accounts.app import accounts_app
 
+from typing import Union, Optional, Any
+
 
 class Confirmation(URLSafeTimedSerializer):
 
@@ -11,7 +13,10 @@ class Confirmation(URLSafeTimedSerializer):
             key = accounts_app.config['SECRET_KEY']
         super(Confirmation, self).__init__(key, salt=realm, **kwargs)
 
-    def loads_http(self, s, max_age=None, return_timestamp=False, salt=None):
+    def loads_http(self, s: Union[str, bytes],
+                   max_age: Optional[int] = None,
+                   return_timestamp: bool = False,
+                   salt: Optional[bytes] = None) -> Any:
         """
         Like `Confirmation.loads`, but raise HTTP exceptions with appropriate
         messages instead of `BadSignature` or `SignatureExpired`.