getEntity() fixes