Fix broken interface for function used for tests