diff --git a/contracts/token/DODOMine.sol b/contracts/token/DODOMine.sol
index c22391d..8734c9f 100644
--- a/contracts/token/DODOMine.sol
+++ b/contracts/token/DODOMine.sol
@@ -132,7 +132,10 @@ contract DODOMine is Ownable {
         poolInfos[pid].allocPoint = _allocPoint;
     }
 
-    function setReward(uint256 _dodoPerBlock) external onlyOwner {
+    function setReward(uint256 _dodoPerBlock, bool _withUpdate) external onlyOwner {
+        if (_withUpdate) {
+            massUpdatePools();
+        }
         dodoPerBlock = _dodoPerBlock;
     }