Error marks in the complete body when method return type is wrong

Currently, the complete body of a method is marked red when the return type of the method is not the specified return type.

This is very annoying because when initially writing a method, everything stays red until you have finished writing. This means that I can't spot errors in the code before having completed the method. I often add a "null" at the end of the method just to make the wavy lines go away.

I think this really needs to be changed. Maybe only the result type specification of the method can be marked or the closing brace?