2 files changed, 46 insertions, 0 deletions

diff --git a/spec/inputs/cond.yue b/spec/inputs/cond.yue
index 3dee99b..d33eec4 100644
--- a/spec/inputs/cond.yue
+++ b/spec/inputs/cond.yue
@@ -182,7 +182,23 @@ m = if 1
 
 else 6
 
+do
+        a
+                :b
+        return if a
+                :b
+        else if c
+                d: e
+        else
+                f: 123
 
+do
+        c
+                d: e
+        if a
+                :b = tb
+        elseif c
+                d: e = tb
 
 nil
 
diff --git a/spec/outputs/cond.lua b/spec/outputs/cond.lua
index 5efa237..8f69e97 100644
--- a/spec/outputs/cond.lua
+++ b/spec/outputs/cond.lua
@@ -269,4 +269,34 @@ if 1 then
 else
         m = 6
 end
+do
+        a({
+                b = b
+        })
+        if a then
+                return {
+                        b = b
+                }
+        else
+                if c then
+                        return {
+                                d = e
+                        }
+                else
+                        return {
+                                f = 123
+                        }
+                end
+        end
+end
+do
+        c({
+                d = e
+        })
+        if a then
+                b = tb.b
+        elseif c then
+                local e = tb.d
+        end
+end
 return nil