From f64a2583b8c313b0c79ee4eec9fd29c0512632e9 Mon Sep 17 00:00:00 2001
From: Hillel Coren <hillelcoren@gmail.com>
Date: Sun, 29 Apr 2018 09:00:22 +0300
Subject: [PATCH] Improve late fee calculation

---
 app/Ninja/Repositories/InvoiceRepository.php | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/app/Ninja/Repositories/InvoiceRepository.php b/app/Ninja/Repositories/InvoiceRepository.php
index e6404ef538e9..8bcd6dce30e3 100644
--- a/app/Ninja/Repositories/InvoiceRepository.php
+++ b/app/Ninja/Repositories/InvoiceRepository.php
@@ -1285,8 +1285,8 @@ class InvoiceRepository extends BaseRepository
         $data = $invoice->toArray();
         $fee = $amount;
 
-        if ($invoice->balance > 0) {
-            $fee += round($invoice->balance * $percent / 100, 2);
+        if ($invoice->getRequestedAmount() > 0) {
+            $fee += round($invoice->getRequestedAmount() * $percent / 100, 2);
         }
 
         $item = [];