Turns out we need the entity type as it's not part of the entity object.