Handle possible inputs for entity_id