Fixed #12 - Dataset.get_entity() now returns the correct type.