Fix function signature mentioned in #2901