correct some codes

correct some codes