Fixed an error with the test file