Added missing include, fixes #4

Thanks @kevwilde for your issue.