diff --git a/auth/v1/msg.proto b/auth/v1/msg.proto
index 34cf9aa..f8649b0 100644
--- a/auth/v1/msg.proto
+++ b/auth/v1/msg.proto
@@ -194,6 +194,8 @@ message TFAReq {
   optional string mobile = 2;
   optional string email = 3;
   optional string reason = 4; // issued jwt reason -> can be login, redeem, withdrawal
+  optional string transaction_id = 5;
+  optional TfaStateEnum state = 6;
 }
 
 // Two Factor Authentication Exchange Response
diff --git a/wallet/v1/msg.proto b/wallet/v1/msg.proto
index 3d58a21..f8f70c3 100644
--- a/wallet/v1/msg.proto
+++ b/wallet/v1/msg.proto
@@ -302,6 +302,7 @@ message InternalTransferAssetReq {
 message InternalTransferAssetRes {
   auth.v1.Recipient recipient = 1;
   double amount = 2;
+  // TODO: Change to Asset
   string transaction_id = 3;
   string transaction_hash = 4;
 }