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