Fix compilation
Merge pull request #9 from vikramrajkumar/upstream

Fix compilation