adding a missing return