fix up the unit tests
Merge pull request #9 from Medium/nick-dns

fix up the unit tests