Fix regression introduced by previous commit