Moved the num rounds mux into the FSM since num rounds is only used there.