fix issue in the previous commit