fixes #1779

Fixed the issue and pushed code for review, please check it.