fix test code error