fix: no need to distributeRewards if balance is 0 (yearn#114)

chriswessels · Nov 29, 2020 · a06f5b2 · a06f5b2
1 parent 99dcc2a
commit a06f5b2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/contracts/BaseStrategy.sol b/contracts/BaseStrategy.sol
@@ -308,7 +308,10 @@ abstract contract BaseStrategy {
      */
     function distributeRewards() internal virtual {
         // Transfer 100% of newly-minted shares awarded to this contract to the rewards address.
-        vault.transfer(rewards, vault.balanceOf(address(this)));
+        uint256 balance = vault.balanceOf(address(this));
+        if (balance > 0) {
+            vault.transfer(rewards, balance);
+        }
     }
 
     /*