fix code removal error