Minor adjustments to the unit tests
Merge pull request #1 from financeCoding/donny-dont-cleanup

Minor adjustments to the unit tests