Fix issue due to previous commit