fixed the minor bug