diff --git a/src/main/scala/leon/plugin/CodeExtraction.scala b/src/main/scala/leon/plugin/CodeExtraction.scala
index 4d3dab3d0f84fa215a588712bfe82e57f63abb51..ae7d8c0e5c90f54afc179b5ee17fbf1ef885db1b 100644
--- a/src/main/scala/leon/plugin/CodeExtraction.scala
+++ b/src/main/scala/leon/plugin/CodeExtraction.scala
@@ -1065,6 +1065,10 @@ trait CodeExtraction extends Extractors {
     case LetVar(_, _, rest) => getReturnedExpr(rest)
     case PBlock(_, rest) => getReturnedExpr(rest)
     case IfExpr(_, then, elze) => getReturnedExpr(then) ++ getReturnedExpr(elze)
+    case MatchExpr(_, cses) => cses.flatMap{
+      case SimpleCase(_, rhs) => getReturnedExpr(rhs)
+      case GuardedCase(_, _, rhs) => getReturnedExpr(rhs)
+    }
     case _ => Seq(expr)
   }