From 9534dcafc6c9e311ebac4b96b3428c04750c7267 Mon Sep 17 00:00:00 2001
From: Yusuke Endoh <mame@ruby-lang.org>
Date: Thu, 22 Aug 2024 16:03:29 +0900
Subject: [PATCH] Fix an error on multi next statements

---
 lib/typeprof/core/graph/box.rb     |  2 +-
 scenario/regressions/multi-next.rb | 13 +++++++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)
 create mode 100644 scenario/regressions/multi-next.rb

diff --git a/lib/typeprof/core/graph/box.rb b/lib/typeprof/core/graph/box.rb
index 8f4d5eec..6d5097e8 100644
--- a/lib/typeprof/core/graph/box.rb
+++ b/lib/typeprof/core/graph/box.rb
@@ -254,7 +254,7 @@ def show
   class EscapeBox < Box
     def initialize(node, genv, a_ret, f_ret)
       super(node)
-      @a_ret = a_ret
+      @a_ret = a_ret.new_vertex(genv, node)
       @f_ret = f_ret
       @f_ret.add_edge(genv, self)
     end
diff --git a/scenario/regressions/multi-next.rb b/scenario/regressions/multi-next.rb
new file mode 100644
index 00000000..74c61e40
--- /dev/null
+++ b/scenario/regressions/multi-next.rb
@@ -0,0 +1,13 @@
+## update
+def check
+  ["str"].map do |x|
+    next x
+    next x
+    1
+  end
+end
+
+## assert
+class Object
+  def check: -> Array[Integer | String]
+end