fixes #1861

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