Fix issue with the previous commit