modify testcode