Improve type annotations on Builder.run()