Fix a missing return