Improved type propagation to stubs