1 files changed, 2 insertions, 2 deletions
diff --git a/src/Language/Fiddle/Compiler/ConsistencyCheck.hs b/src/Language/Fiddle/Compiler/ConsistencyCheck.hs
index 903e6f4..a4f252e 100644
--- a/src/Language/Fiddle/Compiler/ConsistencyCheck.hs
+++ b/src/Language/Fiddle/Compiler/ConsistencyCheck.hs
@@ -77,7 +77,7 @@ deriving instance AdvanceStage S FiddleDecl
 
 instance AdvanceStage S FiddleUnit where
   advanceStage () fu@(FiddleUnit _ decls a) =
-    FiddleUnit (getUnitInterface fu) <$> mapM (advanceStage ()) decls <*> pure a
+    FiddleUnit (Present $ getUnitInterface fu) <$> mapM (advanceStage ()) decls <*> pure a
     where
       getUnitInterface = execWriter . walk_ doWalk
 
@@ -102,7 +102,7 @@ instance AdvanceStage S FiddleUnit where
                 tell (UnitInterface.singleton d)
             | (Just (ImportStatement {importInterface = ii})) <-
                 castTS t ->
-                tell (UnitInterface mempty (dependencies ii))
+                tell (UnitInterface mempty (dependencies (unwrap ii)))
           _ -> return ()
 
       castTS ::