correct some little bugs