Add missing return keyword