Add another missing return