From caa32f8c04887c9930d101082f764ed36a5b63fb Mon Sep 17 00:00:00 2001
From: Adam Izraelevitz
Date: Tue, 14 Mar 2017 12:35:52 -0700
Subject: Small fix

---
 src/main/scala/tutorial/lesson2-working-ir/AnalyzeCircuit.scala | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/main/scala/tutorial')

diff --git a/src/main/scala/tutorial/lesson2-working-ir/AnalyzeCircuit.scala b/src/main/scala/tutorial/lesson2-working-ir/AnalyzeCircuit.scala
index d0e831e9..bdf6f665 100644
--- a/src/main/scala/tutorial/lesson2-working-ir/AnalyzeCircuit.scala
+++ b/src/main/scala/tutorial/lesson2-working-ir/AnalyzeCircuit.scala
@@ -148,9 +148,9 @@ class AnalyzeCircuit extends Transform {
     // Execute the function walkExpression(ledger) on every [[Expression]] in e,
     //  then handle if a [[Mux]].
     e map walkExpression(ledger) match {
-      case Mux(cond, tval, fval, tpe) =>
-        ledger.foundMux
-        e
+      case mux: Mux =>
+        ledger.foundMux()
+        mux
       case notmux => notmux
     }
   }
-- 
cgit v1.2.3