Revert "Fixed StateFigure to InitialStateFigure Exception"

This reverts commit b12c82ae8304816ced48a2d72809c78c4b84f404.