diff --git a/GMERemittance/Module/SendMoney/SendMoneyReceipt/User Interface/View/SendMoneyReceiptViewController.swift b/GMERemittance/Module/SendMoney/SendMoneyReceipt/User Interface/View/SendMoneyReceiptViewController.swift index 616140f5..592f2399 100644 --- a/GMERemittance/Module/SendMoney/SendMoneyReceipt/User Interface/View/SendMoneyReceiptViewController.swift +++ b/GMERemittance/Module/SendMoney/SendMoneyReceipt/User Interface/View/SendMoneyReceiptViewController.swift @@ -78,7 +78,11 @@ class SendMoneyReceiptViewController: UIViewController { // MARK: IBActions @IBAction func done(_ sender: UIButton) { - self.navigationController?.popToRootViewController(animated: true) + if (self.shouldShowCancelAmmendButton) { + self.navigationController?.popToRootViewController(animated: true) + }else { + self.navigationController?.popViewController(animated: true) + } } // MARK: Other Functions