Now fixed the minor bug