Fixes the regression introduced by the previous commit