Add missing include
Merge pull request #7 from spoida/master

Add missing include