diff --git a/auth/v1/msg.proto b/auth/v1/msg.proto
index f8649b0..392f103 100644
--- a/auth/v1/msg.proto
+++ b/auth/v1/msg.proto
@@ -205,10 +205,13 @@ message TFAExRes {
   string qrcode_base64 = 3;
 }
 message CheckTFAReq {
-  InternalIAM iam = 1;
+  // InternalIAM iam = 1;
+  TFAReq req = 1;
   string code = 2;
-  optional string algorithm = 3;
-  optional string reason = 4; // Can be later used as a lookup
+  // optional string algorithm = 3;
+  // optional string reason = 4; // Can be later used as a lookup
+  // optional string transaction_id = 5;
+  // optional TfaStateEnum state = 6;
 }
 
 message TFA {