Add missing include
Merge pull request #54 from twhittock/master

Add missing include