Cleaning test code