fix of test code