cleanup tests code