diff --git a/wallet_message.proto b/wallet_message.proto
index d85564a..cd77759 100644
--- a/wallet_message.proto
+++ b/wallet_message.proto
@@ -126,7 +126,7 @@ message UserBNPLResp {
   BNPLInfo bnpl = 2;
   int32 current_step = 3;
   optional float amount = 4;
-  optional float payment_period = 5;
+  optional PaymentPeriodInfo payment_period = 5;
   optional bool accepted_terms = 6;
   repeated BNPLPayments payments = 7;
   optional string employee_id = 8;
@@ -142,7 +142,7 @@ message UserBNPLReq {
   string employee_id = 4;
   int64 company_id = 5;
   float amount = 7;
-  float payment_period = 8;
+  PaymentPeriodInfo payment_period = 8;
   bool has_agreed_contract = 9;
 }
 message PaymentPeriodInfo {