modify some test code