[feature] fix test