module GraphQL::Schema::Resolver::HasPayloadType

def type(new_type = nil, null: nil)
  if new_type
    payload_type(new_type)
    if !null.nil?
      self.null(null)
    end
  else
    super()
  end
end