Fix unit test file