fixed bug with saving of polarity for same problems in predict