Fix includes in test suite.
Merge pull request #3 from bchretien/master

Minor fix