Updated the test