diff --git a/api/api.proto b/api/api.proto
index 07c0c7e32d5..e95ef1fb651 100644
--- a/api/api.proto
+++ b/api/api.proto
@@ -112,14 +112,14 @@ service Wallet {
     };
   }
 
-  rpc UnfreezeBalance (EmptyMessage) returns (Transaction) {
+  rpc UnfreezeBalance (UnfreezeBalanceContract) returns (Transaction) {
     option (google.api.http) = {
       post: "/wallet/unfreezebalance"
       body: "*"
     };
   }
 
-  rpc WithdrawBalance (EmptyMessage) returns (Transaction) {
+  rpc WithdrawBalance (WithdrawBalanceContract) returns (Transaction) {
     option (google.api.http) = {
       post: "/wallet/withdrawbalance"
       body: "*"
diff --git a/core/Contract.proto b/core/Contract.proto
index 129d6dc6f61..7cc140dd04a 100644
--- a/core/Contract.proto
+++ b/core/Contract.proto
@@ -103,6 +103,15 @@ message DeployContract {
 }
 
 message FreezeBalanceContract {
-  int64 frozen_balance = 1;
-  int64 frozen_duration = 2;
+  bytes owner_address = 1;
+  int64 frozen_balance = 2;
+  int64 frozen_duration = 3;
+}
+
+message UnfreezeBalanceContract {
+  bytes owner_address = 1;
 }
+
+message WithdrawBalanceContract {
+  bytes owner_address = 1;
+}
\ No newline at end of file
diff --git a/core/Tron.proto b/core/Tron.proto
index 854366643e1..0d2b83df809 100644
--- a/core/Tron.proto
+++ b/core/Tron.proto
@@ -126,6 +126,9 @@ message Transaction {
       WitnessUpdateContract = 8;
       ParticipateAssetIssueContract = 9;
       AccountUpdateContract = 10;
+      FreezeBalanceContract = 11;
+      UnfreezeBalanceContract = 12;
+      WithdrawBalanceContract = 13;
       CustomContract = 20;
     }
     ContractType type = 1;