diff --git a/mediapipe/python/BUILD b/mediapipe/python/BUILD
index 085fbc96bc..b6e1ab945f 100644
--- a/mediapipe/python/BUILD
+++ b/mediapipe/python/BUILD
@@ -52,6 +52,7 @@ pybind_extension(
         "@stblib//:stb_image",
         # Type registration.
         "//mediapipe/framework:basic_types_registration",
+        "//mediapipe/framework/formats:body_rig_registration",
         "//mediapipe/framework/formats:classification_registration",
         "//mediapipe/framework/formats:detection_registration",
         "//mediapipe/framework/formats:landmark_registration",
diff --git a/mediapipe/python/solution_base.py b/mediapipe/python/solution_base.py
index 020d163716..ad57ef07c3 100644
--- a/mediapipe/python/solution_base.py
+++ b/mediapipe/python/solution_base.py
@@ -40,6 +40,7 @@ class contains the shared logic among the high-level Solution APIs including
 from mediapipe.calculators.util import logic_calculator_pb2
 from mediapipe.calculators.util import thresholding_calculator_pb2
 from mediapipe.framework import calculator_pb2
+from mediapipe.framework.formats import body_rig_pb2
 from mediapipe.framework.formats import classification_pb2
 from mediapipe.framework.formats import detection_pb2
 from mediapipe.framework.formats import landmark_pb2
@@ -192,6 +193,12 @@ def from_registered_name(registered_name: str) -> 'PacketDataType':
         PacketDataType.PROTO_LIST,
     '::std::vector<::mediapipe::NormalizedRect>':
         PacketDataType.PROTO_LIST,
+    '::mediapipe::Joint':
+        PacketDataType.PROTO,
+    '::mediapipe::JointList':
+        PacketDataType.PROTO,
+    '::std::vector<::mediapipe::JointList>':
+        PacketDataType.PROTO_LIST,
 }