Fixed potential testing issue
Merge pull request #171 from GrahamCampbell/tests

Fixed potential testing issue