Fixed StateFigure to InitialStateFigure Exception

duh