diff --git a/twml/twml/trainers/__init__.py b/twml/twml/trainers/__init__.py
index e6664d9a6..8ae2fd95f 100644
--- a/twml/twml/trainers/__init__.py
+++ b/twml/twml/trainers/__init__.py
@@ -6,5 +6,5 @@ Trainers wrap a
 <https://www.tensorflow.org/versions/master/api_docs/python/tf/estimator/Estimator>`_.
 """
 
-from .trainer import Trainer  # noqa: F401
+from .trainer import Trainer                        # noqa: F401
 from .data_record_trainer import DataRecordTrainer  # noqa: F401