Fix issue introduced with previous commit