fix: setStrategyStatus can be called by strategist.

OriginProtocol · Dec 30, 2024 · 23a885f · 23a885f
1 parent 500b6d3
commit 23a885f
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/contracts/contracts/harvest/OETHHarvesterSimple.sol b/contracts/contracts/harvest/OETHHarvesterSimple.sol
@@ -19,6 +19,14 @@ contract OETHHarvesterSimple is Governable {
     event StrategistSet(address strategist);
     event StrategyStatusSet(address strategy, bool status);
 
+    ////////////////////////////////////////////////////
+    /// --- MODIFIERS
+    ////////////////////////////////////////////////////
+    modifier onlyStrategist() {
+        require(msg.sender == strategist || isGovernor(), "Not strategist");
+        _;
+    }
+
     ////////////////////////////////////////////////////
     /// --- CONSTRUCTOR
     ////////////////////////////////////////////////////
@@ -67,7 +75,7 @@ contract OETHHarvesterSimple is Governable {
     ////////////////////////////////////////////////////
     function setStrategyStatus(address _strategy, bool _status)
         external
-        onlyGovernor
+        onlyStrategist
     {
         isAuthorized[_strategy] = _status;
         emit StrategyStatusSet(_strategy, _status);