Fix test codes.