Update type context after inlining