fix _beforeTokenTransfer issue on MasaSBTDynamic

masa-finance · Nov 7, 2023 · b01f445 · b01f445
1 parent 40796f5
commit b01f445
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/contracts/tokens/MasaSBTDynamic.sol b/contracts/tokens/MasaSBTDynamic.sol
@@ -96,8 +96,7 @@ abstract contract MasaSBTDynamic is MasaSBT {
 
         for (uint256 i = 0; i < _afterMintStates.length; i++) {
             if (
-                keccak256(bytes(_afterMintStates[i])) ==
-                keccak256(bytes(state))
+                keccak256(bytes(_afterMintStates[i])) == keccak256(bytes(state))
             ) {
                 _afterMintStates[i] = _afterMintStates[
                     _afterMintStates.length - 1
@@ -178,10 +177,12 @@ abstract contract MasaSBTDynamic is MasaSBT {
     /// @notice Checks if a token can be minted
     /// @dev Checks if all beforeMintStates are set for the account
     function _beforeTokenTransfer(
-        address,
+        address from,
         address to,
-        uint256
+        uint256 tokenId
     ) internal virtual override {
+        super._beforeTokenTransfer(from, to, tokenId);
+
         if (to != address(0)) {
             if (!allBeforeMintStatesSet(to)) revert NotAllBeforeMintStatesSet();
         }