3 files changed, 3 insertions, 2 deletions
diff --git a/src/comp/driver/rustc.rs b/src/comp/driver/rustc.rs
index 657c098d..a1fdb387 100644
--- a/src/comp/driver/rustc.rs
+++ b/src/comp/driver/rustc.rs
@@ -8,7 +8,7 @@ import middle.trans;
 import middle.resolve;
 import middle.ty;
 import middle.typeck;
-// import middle.typestate_check;
+import middle.typestate_check;
 import util.common;
 
 import std.map.mk_hashmap;
diff --git a/src/comp/middle/typestate_check.rs b/src/comp/middle/typestate_check.rs
index 9b670fc1..46bf6a4c 100644
--- a/src/comp/middle/typestate_check.rs
+++ b/src/comp/middle/typestate_check.rs
@@ -341,7 +341,7 @@ impure fn find_pre_post_item(fn_info enclosing, &item i) -> item {
     case (ast.item_ty(_,_,_,_,_)) {
       ret i;
     }
-    case (ast.item_tag(_,_,_,_)) {
+    case (ast.item_tag(_,_,_,_,_)) {
       ret i;
     }
     case (ast.item_obj(?id, ?o, ?ps, ?di, ?a)) {
diff --git a/src/comp/rustc.rc b/src/comp/rustc.rc
index 98fae38b..952ea7e8 100644
--- a/src/comp/rustc.rc
+++ b/src/comp/rustc.rc
@@ -20,6 +20,7 @@ mod middle {
     mod trans;
     mod ty;
     mod typeck;
+    mod typestate_check;
 }
 
 mod back {