Fix not checking remove argument