Fix a broken test part 2