Fix issue in previous commit