cleanup test code
Merge pull request #2 from AdrianTM/patch-1

cleanup test code