correct minor bug