added done function in both cases

6 years ago · 1013e7b6fc
1 changed files with 5 additions and 1 deletions
--- a/GMERemittance/Module/SendMoney/SendMoneyReceipt/User
+++ b/GMERemittance/Module/SendMoney/SendMoneyReceipt/User
@ -78,7 +78,11 @@ class SendMoneyReceiptViewController: UIViewController {
    
    // MARK: IBActions
    @IBAction func done(_ sender: UIButton) {
+        if (self.shouldShowCancelAmmendButton) {
            self.navigationController?.popToRootViewController(animated: true)
+        }else {
+            self.navigationController?.popViewController(animated: true)
+        }
    }
    // MARK: Other Functions