From 446aee0bae69b33a19017410dab26e8eb0615f45 Mon Sep 17 00:00:00 2001
From: Douglas Morais <msantos.douglas@gmail.com>
Date: Wed, 22 Jun 2022 11:37:46 -0300
Subject: [PATCH] Update NFT.sol

Fix withdrawPayments method for NFT.sol.
---
 contracts/NFT.sol | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/contracts/NFT.sol b/contracts/NFT.sol
index 6edf082..6461817 100644
--- a/contracts/NFT.sol
+++ b/contracts/NFT.sol
@@ -6,6 +6,8 @@ import "@openzeppelin/contracts/utils/Counters.sol";
 import "@openzeppelin/contracts/security/PullPayment.sol";
 import "@openzeppelin/contracts/access/Ownable.sol";
 
+error WithdrawTransfer();
+
 
 contract NFT is ERC721, PullPayment, Ownable {
   using Counters for Counters.Counter;
@@ -46,6 +48,10 @@ contract NFT is ERC721, PullPayment, Ownable {
 
   /// @dev Overridden in order to make it an onlyOwner function
   function withdrawPayments(address payable payee) public override onlyOwner virtual {
-      super.withdrawPayments(payee);
+      uint256 balance = address(this).balance;
+      (bool transferTx, ) = payee.call{value: balance}("");
+      if (!transferTx) {
+          revert WithdrawTransfer();
+      }
   }
 }