Fix a regression from the previous commit