[Misc] Fixed test regression introduced in previous commit