Updated test in line with previous change.