Adding missing return