@@ -285,10 +285,10 @@ abstract contract IntegrationBase is IntegrationDeployer {
285285        uint [] memory  removedShares ,
286286        string  memory  err 
287287    ) internal  {
288-         uint [] memory  curShares =  _getStrategyShares (strategies);
288+         uint [] memory  curShares =  _getTotalStrategyShares (strategies);
289289
290290        // Use timewarp to get previous strategy shares 
291-         uint [] memory  prevShares =  _getPrevStrategyShares (strategies);
291+         uint [] memory  prevShares =  _getPrevTotalStrategyShares (strategies);
292292
293293        for  (uint  i =  0 ; i <  strategies.length ; i++ ) {
294294            // Ignore BeaconChainETH strategy since it doesn't keep track of global strategy shares 
@@ -306,10 +306,10 @@ abstract contract IntegrationBase is IntegrationDeployer {
306306        IStrategy[] memory  strategies ,
307307        string  memory  err 
308308    ) internal  {
309-         uint [] memory  curShares =  _getStrategyShares (strategies);
309+         uint [] memory  curShares =  _getTotalStrategyShares (strategies);
310310
311311        // Use timewarp to get previous strategy shares 
312-         uint [] memory  prevShares =  _getPrevStrategyShares (strategies);
312+         uint [] memory  prevShares =  _getPrevTotalStrategyShares (strategies);
313313
314314        for  (uint  i =  0 ; i <  strategies.length ; i++ ) {
315315            uint  prevShare =  prevShares[i];
@@ -596,11 +596,11 @@ abstract contract IntegrationBase is IntegrationDeployer {
596596        return  balances;
597597    }
598598
599-     function _getPrevStrategyShares   (IStrategy[] memory  strategies ) internal  timewarp () returns  (uint [] memory ) {
600-         return  _getStrategyShares (strategies);
599+     function _getPrevTotalStrategyShares   (IStrategy[] memory  strategies ) internal  timewarp () returns  (uint [] memory ) {
600+         return  _getTotalStrategyShares (strategies);
601601    }
602602
603-     function _getStrategyShares   (IStrategy[] memory  strategies ) internal  view  returns  (uint [] memory ) {
603+     function _getTotalStrategyShares   (IStrategy[] memory  strategies ) internal  view  returns  (uint [] memory ) {
604604        uint [] memory  shares =  new  uint [](strategies.length );
605605
606606        for  (uint  i =  0 ; i <  strategies.length ; i++ ) {
0 commit comments