BUG: correct minor bug.