fix the minor bug