diff --git a/lark-stubs/lark.pyi b/lark-stubs/lark.pyi
index 4b44cc5..ec71f52 100644
--- a/lark-stubs/lark.pyi
+++ b/lark-stubs/lark.pyi
@@ -7,6 +7,7 @@ from typing import (
 from .visitors import Transformer
 from .lexer import Token, Lexer, TerminalDef
 from .tree import Tree
+from .exceptions import UnexpectedInput
 
 _T = TypeVar('_T')
 
@@ -81,7 +82,7 @@ class Lark:
     ):
         ...
 
-    def parse(self, text: str, start: Optional[str] = None, on_error: Callable[[Any], bool] = None) -> Tree:
+    def parse(self, text: str, start: Optional[str] = None, on_error: Callable[[UnexpectedInput], bool] = None) -> Tree:
         ...
 
     @classmethod