Fixed broken test after merging in master.