Fixed test broken by previous commit  (I am too lazy for rebase here)