bug on removing units