Fixed the entity step endpoint