Correclty mimic yellowhiggs function signature