fix test code bug