Highlight face in edit mode stable

Fixed bug that was causing issues trying to highlight faces when in edit
mode.