diff --git a/stubs/psycopg2/psycopg2/_range.pyi b/stubs/psycopg2/psycopg2/_range.pyi
index 131196c95..0b3247455 100644
--- a/stubs/psycopg2/psycopg2/_range.pyi
+++ b/stubs/psycopg2/psycopg2/_range.pyi
@@ -33,7 +33,7 @@ class Range:
     def __ge__(self, other: Range) -> bool: ...
 
 def register_range(
-    pgrange: str, pyrange: str | Range, conn_or_curs: connection | cursor, globally: bool = False
+    pgrange: str, pyrange: str | type[Range], conn_or_curs: connection | cursor, globally: bool = False
 ) -> RangeCaster: ...
 
 class RangeAdapter:
@@ -51,7 +51,12 @@ class RangeCaster:
     typecaster: Any
     array_typecaster: Any
     def __init__(
-        self, pgrange: str | RangeAdapter, pyrange: str | Range, oid: int, subtype_oid: int, array_oid: int | None = None
+        self,
+        pgrange: str | type[RangeAdapter],
+        pyrange: str | type[Range],
+        oid: int,
+        subtype_oid: int,
+        array_oid: int | None = None,
     ) -> None: ...
     @overload
     def parse(self, s: None, cur: cursor | None = None) -> None: ...