fix: update a unit test