A bit of cleaning up in the test file.