Fix unit test includes.