fix ut issue