correct some bug