fix a bug in ut