Updating a unit test