fix cleaning in test