Tree

1 files changed, 2 insertions, 2 deletions

diff --git a/src/Fun/Compiler.hs b/src/Fun/Compiler.hs
index b53ff73..46cfe13 100644
--- a/src/Fun/Compiler.hs
+++ b/src/Fun/Compiler.hs
@@ -23,8 +23,8 @@ traceTree c ts = foldr join "" (map (traceBranch c) ts)
 genTrace :: [Trace] -> String
 genTrace ts = "Trace of expectance:\n" ++ traceTree 0 ts
 
-parse :: String -> Either String Program -- TODO: Should be tree
-parse file = case program file of
+parse :: String -> Either String Tree
+parse file = case tree file of
   Left a -> Left $ "Parse fault!\n" ++ case a of
     State [] Nothing    -> "No context available"
     State t  Nothing    -> genTrace t