assert comparison with await func is parsed incorrectly

[mischab]

Hey, when using await in an assert comparison the output from the parser is incorrect, see example:

assert await a(b) == c

resolves to: (module (assert_statement (await (comparison_operator (call function: (identifier) arguments: (argument_list (identifier))) (identifier)))))