FIx bug on remove method of model.