Make decorate helper method handle collections