fix test error