correct minor bugs