Fix problem where multi-line function arguments are not highlighted properly.