Fix-up type inference for function literals whose returns have been modified.