add basic cleanup to test