switch enum

2 files changed, 2 insertions, 1 deletions

diff --git a/src/fold.cff b/src/fold.cff
index 476e860..a2c95b8 100644
--- a/src/fold.cff
+++ b/src/fold.cff
@@ -113,7 +113,7 @@ extern fn fold(ex *Expr) bool {
       numcast(ex, ex.ty);
       return #t;
 
-   case :StrLit, :NullLit;
+   case :StrLit, :NullLit, :EnumIni;
       return #t;
 
    case :UnOp;
diff --git a/src/parse.cff b/src/parse.cff
index 7ce605c..a2071f8 100644
--- a/src/parse.cff
+++ b/src/parse.cff
@@ -1808,6 +1808,7 @@ fn pstiswitch(P *Parser, loc Loc, ex Expr) Stmt {
       }
       let es Vec<Expr> = {};
       do {
+         P.targty = ex.ty;
          let e = parseexpr(P);
          if !fold(&e) {
             err(P, e.loc, "expected compile-time expression");