Fixed unit tests.
Fixed unit tests.
Merge pull request #5 from AlphaGit/fix-unit-tests

Fix unit tests