BUG: fix compile error, see #2
Merge pull request #3 from fedorov/2-add-missing-include

BUG: fix compile error, see #2