Fix some unit tests to properly test code
Merge pull request #8 from pmaidens/test-fix

Fix some unit tests to properly test the code