fix ut

fix ut