Fix broken build
Merge pull request #141 from keyurdg/master

Fix broken build