Fix unit test project