added fix for arrow and line highlighting