Fix includes
Merge pull request #3 from simonster/fix-include

Fix includes