Update Unipool.sol

Add withdraw and stake requires
Synthetixio · Feb 10, 2020 · 7d1ffb2 · 7d1ffb2
1 parent 35dfa7c
commit 7d1ffb2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/contracts/Unipool.sol b/contracts/Unipool.sol
@@ -85,11 +85,13 @@ contract Unipool is LPTokenWrapper, IRewardDistributionRecipient {
     }
 
     function stake(uint256 amount) public updateReward(msg.sender) {
+        require(amount > 0, "Cannot stake 0");
         super.stake(amount);
         emit Staked(msg.sender, amount);
     }
 
     function withdraw(uint256 amount) public updateReward(msg.sender) {
+        require(amount > 0, "Cannot withdraw 0");
         super.withdraw(amount);
         emit Withdrawn(msg.sender, amount);
     }