adding missing return