Fix test.
Fix includes.