diff --git a/src/Language/Python/Syntax.hs b/src/Language/Python/Syntax.hs
index 63e8df965..dc875499a 100644
--- a/src/Language/Python/Syntax.hs
+++ b/src/Language/Python/Syntax.hs
@@ -207,7 +207,7 @@ argumentList :: Assignment
 argumentList = makeTerm <$> symbol ArgumentList <*> children (many expression)
 
 withStatement :: Assignment
-withStatement = symbol WithStatement >>= \ loc -> children (mk loc <$> some with)
+withStatement = mk <$> symbol WithStatement <*> children (some with)
   where
     mk _ [child] = child
     mk l children = makeTerm l children