I was looking at the test and the implementation.
It looks like with this implementation we'll ignore the return from the function. So we won't show it as red if, for example suspension() function throws an exception. Am I right or I am missing anything here?

Nevertheless, I don't think I have a better idea how to workaround this, so this solution is still better than the current behavior(at the moment it's always red if the last suspending function call contains suspending function invocations).