Small fix

1 files changed, 3 insertions, 3 deletions

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
     }
   }