Add missing set