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