fixed small bug - able to restart the test once it has been run once