add the missing return