Modifying test code