Fix regression introduced in previous commit