fix bug in ut