Missed an include :(
Merge pull request #16 from r0mai/master

Missed an include :(