A minor bug fixed