diff --git a/byg/src/DependencyGenerator.hs b/byg/src/DependencyGenerator.hs
index b71ea6f..fea48b9 100644
--- a/byg/src/DependencyGenerator.hs
+++ b/byg/src/DependencyGenerator.hs
@@ -81,10 +81,10 @@ mapDepGenM f input = genDependencyM $ \target -> do
   put top'
   pure (makeDependency input (MapComp res) target)
 
-mapDepGenM_ :: (Token a -> DepGenM ()) -> Token [a] -> DepGenM ()
+mapDepGenM_ :: (Token a -> DepGenM ()) -> Token [a] -> DepGenM' ()
 mapDepGenM_ f input = do
   _ <- mapDepGenM f input
-  pure NoToken
+  pure ()
 
 filterDepGenM :: (Token a -> DepGenM Bool) -> Token [a] -> DepGenM [a]
 filterDepGenM f input = do