Fix unit test to use the new tt files