cleanup a unit test to test for the right thing