fix assertion in ut