Fix includes in tests.