Small tweak to wizard