Stops highlighting after folded block

[Gabriel-p]

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.