Fixed my issue, have to click on file --> Remove State Return, and then you can redo the State part from scratch again, unlike the button that says to redo the state.
View solution in original post
