diff --git a/python/src/trade/mod.rs b/python/src/trade/mod.rs
index 62b9fd750..982b7b263 100644
--- a/python/src/trade/mod.rs
+++ b/python/src/trade/mod.rs
@@ -25,6 +25,7 @@ pub(crate) fn register_types(parent: &PyModule) -> PyResult<()> {
     parent.add_class::<types::OrderChargeItem>()?;
     parent.add_class::<types::OrderChargeDetail>()?;
     parent.add_class::<types::OrderDetail>()?;
+    parent.add_class::<types::BalanceType>()?;
     parent.add_class::<types::EstimateMaxPurchaseQuantityResponse>()?;
 
     parent.add_class::<context::TradeContext>()?;