Remove method fix