ChangeFunctionReturnTypeFix modified

Continuation of: https://github.com/JetBrains/kotlin/pull/245