diff --git a/pymc/model/core.py b/pymc/model/core.py
index afbd9dc039..f6c69b27bd 100644
--- a/pymc/model/core.py
+++ b/pymc/model/core.py
@@ -1171,8 +1171,8 @@ def set_data(
 
     def register_rv(
         self,
-        rv_var,
-        name,
+        rv_var: TensorVariable,
+        name: str,
         *,
         observed=None,
         total_size=None,
@@ -1180,7 +1180,7 @@ def register_rv(
         default_transform=UNSET,
         transform=UNSET,
         initval=None,
-    ):
+    ) -> TensorVariable:
         """Register an (un)observed random variable with the model.
 
         Parameters