diff --git a/mediapipe/python/BUILD b/mediapipe/python/BUILD
index 085fbc96b..b6e1ab945 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 020d16371..ad57ef07c 100644
--- a/mediapipe/python/solution_base.py
+++ b/mediapipe/python/solution_base.py
@@ -40,6 +40,7 @@ from mediapipe.calculators.util import landmarks_smoothing_calculator_pb2
 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 @@ NAME_TO_TYPE: Mapping[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,
 }