update some test code