fixed a problem about remove code.