fix: use a single deallocation queue

Layr-Labs · Oct 15, 2024 · fc1c22f · fc1c22f
1 parent 4fc6339
commit fc1c22f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/contracts/core/AllocationManager.sol b/src/contracts/core/AllocationManager.sol
@@ -160,6 +160,10 @@ contract AllocationManager is
                 // Calculate the effectTimestamp for the modification
                 if (info.pendingDiff < 0) {
                     info.effectTimestamp = uint32(block.timestamp) + DEALLOCATION_DELAY;
+
+                    // Add the operatorSet to the modification queue and update the allocation
+                    // in storage
+                    modificationQueue[msg.sender][allocation.strategy].pushBack(operatorSetKey);
                 } else if (info.pendingDiff > 0) {
                     info.effectTimestamp = uint32(block.timestamp) + operatorAllocationDelay;
 
@@ -168,10 +172,6 @@ contract AllocationManager is
                     info.encumberedMagnitude = _addInt128(info.encumberedMagnitude, info.pendingDiff);
                     require(info.encumberedMagnitude <= maxMagnitude, InsufficientAllocatableMagnitude());
                 }
-
-                // Add the operatorSet to the modification queue and update the allocation
-                // in storage
-                modificationQueue[msg.sender][allocation.strategy].pushBack(operatorSetKey);
                 _updateMagnitudeInfo({
                     operator: msg.sender,
                     strategy: allocation.strategy,