fix wizard tests