Minor fix after #421

Thanks @gmetais!