Fixed the test file