small modification to the test file