Amend unit test for change in behaviour