Fixing node highlighting on page refresh.