fix test cleanup