Fix for highlighting of edges