Fixing unit test issue
Merge pull request #1 from DevMonkCom/fix/unit-test

Fixing unit test issue
Thanks @DevMonkCom