/** * @param \Magento\Sales\Model\Order\Invoice $invoice * @return $this */ public function collect(\Magento\Sales\Model\Order\Invoice $invoice) { $invoice->setFee(0); $invoice->setBaseFee(0); $amount = $invoice->getOrder()->getFee(); $invoice->setFee($amount); $amount = $invoice->getOrder()->getBaseFee(); $invoice->setBaseFee($amount); $invoice->setGrandTotal($invoice->getGrandTotal() + $invoice->getFee()); $invoice->setBaseGrandTotal($invoice->getBaseGrandTotal() + $invoice->getFee()); return $this; }