Changed the way the test works