diff --git a/test/refactor/extract.py b/test/refactor/extract.py
index f4967ded..312aced8 100644
--- a/test/refactor/extract.py
+++ b/test/refactor/extract.py
@@ -1,6 +1,6 @@
 # --- simple
 def test():
-    #? 25 a
+    #? 35 a
     return test(100, (30 + b, c) + 1)
 
 # +++
@@ -8,6 +8,18 @@ def test():
     a = (30 + b, c) + 1
     return test(100, a)
 
+
+# --- simple #2
+def test():
+    #? 25 a
+    return test(100, (30 + b, c) + 1)
+
+# +++
+def test():
+    a = 30 + b
+    return test(100, (a, c) + 1)
+
+
 # --- multiline
 def test():
     #? 30 x
@@ -19,3 +31,17 @@ def test():
                             + 1)
     return test(1, x
 )
+
+
+# --- multiline #2
+def test():
+    #? 25 x
+    return test(1, (30 + b, c) 
+                            + 1)
+# +++
+def test():
+    x = 30 + b
+    return test(1, (x, c) 
+                            + 1)
+
+