Staking Router 1.5

contracts/0.4.24/test_helpers/MinFirstAllocationStrategyConsumerMockLegacyVersion.sol

@@ -12,8 +12,7 @@ contract MinFirstAllocationStrategyConsumerMockLegacyVersion {
         uint256[] memory capacities,
         uint256 maxAllocationSize
     ) public pure returns (uint256 allocated, uint256[] memory newAllocations) {
-        allocated = MinFirstAllocationStrategy.allocate(allocations, capacities, maxAllocationSize);
-        newAllocations = allocations;
+        (allocated, newAllocations)  = MinFirstAllocationStrategy.allocate(allocations, capacities, maxAllocationSize);
     }
 
     function allocateToBestCandidate(

contracts/0.4.24/test_helpers/NodeOperatorsRegistryMock.sol

@@ -143,6 +143,10 @@ contract NodeOperatorsRegistryMock is NodeOperatorsRegistry {
         _setContractVersion(_newBaseVersion);
     }
 
+    function testing_setRewardDistributionStatus(RewardDistributionState _state) external {
+        _updateRewardDistributionState(_state);
+    }
+
     function testing_resetRegistry() external {
         uint256 totalOperatorsCount = TOTAL_OPERATORS_COUNT_POSITION.getStorageUint256();
         TOTAL_OPERATORS_COUNT_POSITION.setStorageUint256(0);