Multiline function args higlighted incorrectly

When I do:
```
#@ lib = struct.make(
#@ arg1=arg1,
#@ arg2=arg2
#@ )
```
the last three lines are being highlighted as if they were comments.

Using version `0.0.2` and vscode `1.45.1` (linux)