fix(RingBufferLib): use wrap helper function

pooltogether · Oct 6, 2021 · aa9c4bd · aa9c4bd
1 parent 03e6514
commit aa9c4bd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/contracts/libraries/RingBufferLib.sol b/contracts/libraries/RingBufferLib.sol
@@ -17,7 +17,7 @@ library RingBufferLib {
         uint256 _amount,
         uint256 _cardinality
     ) internal pure returns (uint256) {
-        return (_index + _cardinality - _amount) % _cardinality;
+        return wrap(_index + _cardinality - _amount, _cardinality);
     }
 
     /// @notice Returns the index of the last recorded TWAB
@@ -33,14 +33,14 @@ library RingBufferLib {
             return 0;
         }
 
-        return (_nextAvailableIndex + _cardinality - 1) % _cardinality;
+        return wrap(_nextAvailableIndex + _cardinality - 1, _cardinality);
     }
 
     function nextIndex(uint256 _currentIndex, uint256 _cardinality)
         internal
         pure
         returns (uint256)
     {
-        return (_currentIndex + 1) % _cardinality;
+        return wrap(_currentIndex + 1, _cardinality);
     }
 }