diff --git a/contracts/vault/Flows.sol b/contracts/vault/Flows.sol
index e9be256..46bcfb3 100644
--- a/contracts/vault/Flows.sol
+++ b/contracts/vault/Flows.sol
@@ -10,7 +10,7 @@ struct Flow {
 }
 
 library Flows {
-  function _totalAt(
+  function totalAt(
     Flow memory flow,
     Timestamp timestamp
   ) internal pure returns (int128) {
diff --git a/contracts/vault/VaultBase.sol b/contracts/vault/VaultBase.sol
index 5bf0487..806a1c8 100644
--- a/contracts/vault/VaultBase.sol
+++ b/contracts/vault/VaultBase.sol
@@ -51,7 +51,7 @@ abstract contract VaultBase {
     if (account.flow.rate != TokensPerSecond.wrap(0)) {
       Lock memory lock = _locks[controller][fund];
       Timestamp end = Timestamps.earliest(timestamp, lock.expiry);
-      int128 accumulated = account.flow._totalAt(end);
+      int128 accumulated = account.flow.totalAt(end);
       if (accumulated >= 0) {
         account.designated += uint128(accumulated);
       } else {
@@ -235,7 +235,7 @@ abstract contract VaultBase {
     Lock memory lock
   ) private pure {
     if (account.flow.rate < TokensPerSecond.wrap(0)) {
-      uint128 outgoing = uint128(-account.flow._totalAt(lock.maximum));
+      uint128 outgoing = uint128(-account.flow.totalAt(lock.maximum));
       require(outgoing <= account.available, InsufficientBalance());
     }
   }