minor fix
update unittest