Quick bugfix to wizard