diff --git a/libyang/context.py b/libyang/context.py
index f9bd5a5..2db9896 100644
--- a/libyang/context.py
+++ b/libyang/context.py
@@ -371,7 +371,7 @@ def find_path(
         self,
         path: str,
         output: bool = False,
-        root_node: Optional["libyang.SNode"] = None,
+        root_node: Optional[SNode] = None,
     ) -> Iterator[SNode]:
         if self.cdata is None:
             raise RuntimeError("context already destroyed")
@@ -404,9 +404,9 @@ def find_path(
     def find_jsonpath(
         self,
         path: str,
-        root_node: Optional["libyang.SNode"] = None,
+        root_node: Optional[SNode] = None,
         output: bool = False,
-    ) -> Optional["libyang.SNode"]:
+    ) -> Optional[SNode]:
         if root_node is not None:
             ctx_node = root_node.cdata
         else: