diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
index 6e8c261b..a8a407fd 100644
--- a/api/onnx_web/diffusers/utils.py
+++ b/api/onnx_web/diffusers/utils.py
@@ -133,7 +133,7 @@ def expand_prompt(
 
 
 def get_tokens_from_prompt(
-    prompt: str, pattern: Pattern[str]
+    prompt: str, pattern: Pattern
 ) -> Tuple[str, List[Tuple[str, float]]]:
     """
     TODO: replace with Arpeggio