fix test code compile error