fixed the tests for the new files.