diff --git a/app/models/repayment.py b/app/models/repayment.py index 785c27b..871da47 100644 --- a/app/models/repayment.py +++ b/app/models/repayment.py @@ -51,8 +51,8 @@ class Repayment(db.Model): @classmethod def create_repayment(cls, repayment_data): logger.info(f"create_repayment data: {repayment_data}") - if repayment_data.LoanStatus not in [LoanStatus.ACTIVE, LoanStatus.START_REPAY]: - raise ValueError(f"Repayment cannot be processed. Loan status: ({repayment_data.LoanStatus})") + if repayment_data["LoanStatus"] not in [LoanStatus.ACTIVE, LoanStatus.START_REPAY]: + raise ValueError(f"Repayment cannot be processed. Loan status: ({repayment_data["LoanStatus"]})") repayment = cls( customer_id=repayment_data.customerId,