fix missing includes
Merge pull request #1 from floe/master

fix missing includes