Changed the remove unit part