Fix highlighting of arrow box