fixed to works on 0.8.x
Merge pull request #5 from arunoda/patch-1

fixed to works on 0.8.x