Fix previous commit issue