Correct some mistake