amend unit test