Modify test to match the new behaviour