feat: allow anyone to claim (#10)

yearn · Aug 3, 2023 · 60e838d · 60e838d
1 parent b9231f2
commit 60e838d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/contracts/VestingEscrowSimple.vy b/contracts/VestingEscrowSimple.vy
@@ -139,7 +139,8 @@ def claim(beneficiary: address = msg.sender, amount: uint256 = MAX_UINT256):
     @param beneficiary Address to transfer claimed tokens to
     @param amount Amount of tokens to claim
     """
-    assert msg.sender == self.recipient  # dev: not recipient
+    recipient: address = self.recipient
+    assert msg.sender == recipient or recipient == beneficiary # dev: not authorized
 
     claim_period_end: uint256 = min(block.timestamp, self.disabled_at)
     claimable: uint256 = min(self._unclaimed(claim_period_end), amount)