Updating wizard tests

Removing containing div from markup and updating jQuery selection logic