Fixed the minor bug