Fix a regression due to the previous commit