Change the return type to match with that of ranker class