fix minor-bug