Added a missing return