Add missing return value