Rewrite type annotations with typing.overload