diff --git a/auth/v1/msg.proto b/auth/v1/msg.proto
index d09aeac..15112bf 100644
--- a/auth/v1/msg.proto
+++ b/auth/v1/msg.proto
@@ -210,7 +210,13 @@ message CheckTFAReq {
 }
 
 message TFA {
-  string key = 1;
+  OtpStateEnum state = 1;
   int64 code = 2;
   string transaction_id = 3;
+}
+
+enum OtpStateEnum {
+  OPT_STATE_ENUM_UNSPECIFIED = 0;
+  LOGIN = 1;
+  PAYMENT = 2;
 }
\ No newline at end of file