- Update unit test