Fix regression introduced by the previous commit