Stops highlighting after folded block

Check the following example

```
test_word = 'aa'

if True:
    print(
        test_word)

print(test_word)
```

The last `test_word` is properly highlighted when the `if` block es expanded, but not anymore when it is folded.