diff --git a/tests/attr/models/test_pytext.py b/tests/attr/models/test_pytext.py
index 0219e7e1c2..2e7c47ce6b 100644
--- a/tests/attr/models/test_pytext.py
+++ b/tests/attr/models/test_pytext.py
@@ -20,7 +20,7 @@
     from pytext.config.doc_classification import ModelInputConfig, TargetConfig
     from pytext.config.field_config import FeatureConfig, WordFeatConfig
     from pytext.data import CommonMetadata
-    from pytext.data.doc_classification_data_handler import (  # @manual=//pytext:main_lib
+    from pytext.data.doc_classification_data_handler import (  # @manual=//pytext:main_lib  # noqa
         DocClassificationDataHandler,
     )
     from pytext.data.featurizer import SimpleFeaturizer