diff --git a/src/Data/Syntax/Assignment.hs b/src/Data/Syntax/Assignment.hs
index 9baea72e4..0b5b3dac3 100644
--- a/src/Data/Syntax/Assignment.hs
+++ b/src/Data/Syntax/Assignment.hs
@@ -249,7 +249,7 @@ data AssignmentState term = AssignmentState
   }
   deriving (Eq, Show)
 
-makeState :: Source.Source -> [Cofree [] (Record (node ': Location))] -> AssignmentState (Cofree [] (Record (node ': Location)))
+makeState :: Source.Source -> [term] -> AssignmentState term
 makeState source nodes = AssignmentState 0 (Info.SourcePos 1 1) source nodes