Add a missing return