Bug fix for polarity assignment