Add missing return payload