diff --git a/pts/model/transformer/transformer_estimator.py b/pts/model/transformer/transformer_estimator.py
index 46d165e..feaed5a 100644
--- a/pts/model/transformer/transformer_estimator.py
+++ b/pts/model/transformer/transformer_estimator.py
@@ -26,6 +26,7 @@ from pts.transform import (
     VstackFeatures,
     SetFieldIfNotPresent,
     TargetDimIndicator,
+    SetField,
 )
 from pts.feature import (
     TimeFeature,