diff --git a/plugins/flytekit-spark/flytekitplugins/spark/pyspark_transformers.py b/plugins/flytekit-spark/flytekitplugins/spark/pyspark_transformers.py
index c957ebc34c..bcbdd2b632 100644
--- a/plugins/flytekit-spark/flytekitplugins/spark/pyspark_transformers.py
+++ b/plugins/flytekit-spark/flytekitplugins/spark/pyspark_transformers.py
@@ -1,11 +1,11 @@
 from typing import Type
 
-from flytekit import Blob, BlobMetadata, BlobType, FlyteContext, Literal, LiteralType, Scalar, lazy_module
+from pyspark.ml import PipelineModel
+
+from flytekit import Blob, BlobMetadata, BlobType, FlyteContext, Literal, LiteralType, Scalar
 from flytekit.core.type_engine import TypeEngine
 from flytekit.extend import TypeTransformer
 
-from pyspark.ml import PipelineModel
-
 
 class PySparkPipelineModelTransformer(TypeTransformer[PipelineModel]):
     _TYPE_INFO = BlobType(format="binary", dimensionality=BlobType.BlobDimensionality.MULTIPART)