modified test code