fixed removal of unit by clicking remove icon