Fix find_by_entity_ids to have the same signature as before