Fix build failure
Merge pull request #21 from NiteshKant/master

Fix build failure