diff --git a/apps/account/models.py b/apps/account/models.py
index e67fef49..96a4916c 100644
--- a/apps/account/models.py
+++ b/apps/account/models.py
@@ -436,7 +436,7 @@ class User(AbstractUser):
                 result.append(item.id)
         return set(result)
 
-    def set_roles(self, ids: List(int)):
+    def set_roles(self, ids: List[int]):
         """
         Set user roles
         :param ids: list of role ids