Quick cleanup for test