Tweak remove code