Add missing return values