Fix error
Merge pull request #214 from rajthilakmca/0.5

Fix error