diff --git a/efficientnet_image_embedding.py b/efficientnet_image_embedding.py
index ecd47dd..bb62091 100644
--- a/efficientnet_image_embedding.py
+++ b/efficientnet_image_embedding.py
@@ -23,7 +23,7 @@ import numpy
 from towhee.operator import Operator
 
 
-class EfficientnetImageEmbeddingOperator(Operator):
+class EfficientnetImageEmbedding(Operator):
     """
     Embedding extractor using efficientnet.
     Args: