Fix the regression introduced by the previous commit