Updated the test file