diff --git a/element_deeplabcut/model.py b/element_deeplabcut/model.py
index 09269f8..d4faa56 100644
--- a/element_deeplabcut/model.py
+++ b/element_deeplabcut/model.py
@@ -776,7 +776,7 @@ def get_trajectory(cls, key: dict, body_parts: list = "all") -> pd.DataFrame:
         df = None
         for body_part in body_parts:
             x_pos, y_pos, z_pos, likelihood = (
-                cls.BodyPartPosition & {"body_part": body_part}
+                cls.BodyPartPosition & key & {"body_part": body_part} #TR23: added key restraint!
             ).fetch1("x_pos", "y_pos", "z_pos", "likelihood")
             if not z_pos:
                 z_pos = np.zeros_like(x_pos)