Little change to CMSPluginBase.save_model