Correcting a few things in the unit test
Merge pull request #4 from chadlung/master

Correcting a few things in the unit test