diff --git a/spacy/_ml.py b/spacy/_ml.py
index 1cd83563e..ffa2951ce 100644
--- a/spacy/_ml.py
+++ b/spacy/_ml.py
@@ -517,7 +517,9 @@ def build_text_classifier(nr_class, width=64, **cfg):
             model = (
                 SpacyVectors
                 >> flatten_add_lengths
-                >> with_getitem(0, LN(Affine(width, 300)))
+                >> with_getitem(0,
+                    Affine(width, 300)
+                )
                 >> ParametricAttention(width)
                 >> Pooling(sum_pool)
                 >> Residual(ReLu(width, width)) ** 2