Same here, no need to copy this class we can make it work for both wizards