fix test codes