Syntax coloring become broken after use of type annotation #662

Any time type annotation used this lead to some weird behavior of coloring syntax.
See color of keywords for this example:

And the correct version:

In more complex cases this also broke code collapsing completely.

Environment data

VS Code version:
Python Extension version:
Python Version: 3.5, 3.6
OS and version: Windows 7, Windows 10