Fix compilation
Merge pull request #9 from alanpearson/master

Fix compilation