fix a bug of previous commit