Fix issue with previous commit