From ad1905a99ddd6073afa5787e16a58e661ed8ee0b Mon Sep 17 00:00:00 2001
From: Jason Zaugg <jzaugg@gmail.com>
Date: Mon, 16 Oct 2017 07:41:41 -0500
Subject: Add missing recursion to branchWithAssign

---
 src/main/scala/scala/async/internal/AnfTransform.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/scala/scala/async/internal/AnfTransform.scala b/src/main/scala/scala/async/internal/AnfTransform.scala
index 93297f7..0085554 100644
--- a/src/main/scala/scala/async/internal/AnfTransform.scala
+++ b/src/main/scala/scala/async/internal/AnfTransform.scala
@@ -118,7 +118,7 @@ private[async] trait AnfTransform {
                     case MatchEnd(ld) =>
                       deriveLabelDef(ld, branchWithAssign)
                     case blk @ Block(thenStats, thenExpr) =>
-                      treeCopy.Block(blk, thenStats, typedAssign(thenExpr)).setType(definitions.UnitTpe)
+                      treeCopy.Block(blk, thenStats, branchWithAssign(thenExpr)).setType(definitions.UnitTpe)
                     case _ =>
                       typedAssign(t)
                   }
-- 
cgit v1.2.3