Fixed unit test
Merge pull request #14 from rmansoor/master

Fix unit test