Fix nodes management

ambrosus · Sep 12, 2024 · 19ac273 · 19ac273
1 parent 935b3ae
commit 19ac273
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/contracts/staking/liquid/LiquidNodesManager.sol b/contracts/staking/liquid/LiquidNodesManager.sol
@@ -54,6 +54,10 @@ contract LiquidNodesManager is UUPSUpgradeable, AccessControlUpgradeable {
     // POOL METHODS
 
     function stake() external payable onlyRole(POOL_ROLE) {
+        // retire node if maxNodesCount was changed
+        if (nodes.length > maxNodesCount) {
+            _retireNode();
+        }
         // try to onboard new node
         _requestNodeCreation();
     }
@@ -64,6 +68,8 @@ contract LiquidNodesManager is UUPSUpgradeable, AccessControlUpgradeable {
             _retireNode();
         }
         transferViaCall(payable(msg.sender), amount);
+        // Try to onboard new node in case if maxNodesCount was changed
+        _requestNodeCreation();
     }
     // IStakeManager impl