Cleanup tests code