fix build break
Merge pull request #251 from manishsing/master

fix build break