diff --git a/libs/mocks/src/pools.rs b/libs/mocks/src/pools.rs
index cd6baa52d1..5660251a24 100644
--- a/libs/mocks/src/pools.rs
+++ b/libs/mocks/src/pools.rs
@@ -77,7 +77,9 @@ pub mod pallet {
 			register_call!(move |(a, b, c)| f(a, b, c));
 		}
 
-		pub fn mock_bench_default_investment_id(f: impl Fn(T::TrancheCurrency) + 'static) {
+		pub fn mock_bench_default_investment_id(
+			f: impl Fn(T::PoolId) -> T::TrancheCurrency + 'static + 'static,
+		) {
 			register_call!(f);
 		}
 	}