Fix ERC20Snapshot#_beforeTokenTransfer (#2328)

5 years ago · 6d5a73815d
parent d1f336d8fd
commit 6d5a73815d
1 changed files with 6 additions and 2 deletions
--- a/contracts/token/ERC20/ERC20Snapshot.sol
+++ b/contracts/token/ERC20/ERC20Snapshot.sol
@ -110,8 +110,12 @@ abstract contract ERC20Snapshot is ERC20 {
    function _beforeTokenTransfer(address from, address to, uint256 amount) internal virtual override {
      super._beforeTokenTransfer(from, to, amount);

-      if (from == address(0) || to == address(0)) {
-        // mint or burn
+      if (from == address(0)) {
+        // mint
+        _updateAccountSnapshot(to);
+        _updateTotalSupplySnapshot();
+      } else if (to == address(0)) {
+        // burn
        _updateAccountSnapshot(from);
        _updateTotalSupplySnapshot();
      } else {