class ZuoraConnect::Login

def client(id = self.default_entity)
  use_entity_name = self.identifier_to_ids.keys.include?(id)
  # Translate entity name to entity id
  id = self.identifier_to_ids[id] if use_entity_name
  client = id.blank? ? self.clients[self.default_entity] : self.clients[id]
  if client
    client.entity_header_type = use_entity_name ? :entity_name : :entity_id
  end
  client
end