modify some test code. but without any test
this should be tested afterwards