Highlight polygon fix