Fixed unit tests
Merge pull request #10 from immersivelabs/fix/unit-tests

[RFC] fix unit tests