correct minor bug
correct minor bug