Fix unit test crash.
Merge pull request #13 from gamechanger/jh-fix-unit-test

Fix unit test crash.