add ut, fix