diff --git a/include/pingcap/kv/internal/type_traits.h b/include/pingcap/kv/internal/type_traits.h
index af9e5d45..5d43b5e2 100644
--- a/include/pingcap/kv/internal/type_traits.h
+++ b/include/pingcap/kv/internal/type_traits.h
@@ -46,6 +46,7 @@ PINGCAP_DEFINE_TRAITS(kvrpcpb, ResolveLock, KvResolveLock)
 PINGCAP_DEFINE_TRAITS(kvrpcpb, PessimisticRollback, KVPessimisticRollback)
 PINGCAP_DEFINE_TRAITS(kvrpcpb, TxnHeartBeat, KvTxnHeartBeat)
 PINGCAP_DEFINE_TRAITS(kvrpcpb, CheckSecondaryLocks, KvCheckSecondaryLocks)
+PINGCAP_DEFINE_TRAITS(disaggregated, EstablishDisaggTask, EstablishDisaggTask)
 PINGCAP_DEFINE_TRAITS(disaggregated, CancelDisaggTask, CancelDisaggTask)
 PINGCAP_DEFINE_TRAITS(coprocessor, , Coprocessor)
 PINGCAP_DEFINE_TRAITS(mpp, DispatchTask, DispatchMPPTask)