Update unit test

update