Fix includes in testing