diff --git a/transform_image_operator_template.yaml b/transform_image_operator_template.yaml
index 13e00eb..835b1c1 100644
--- a/transform_image_operator_template.yaml
+++ b/transform_image_operator_template.yaml
@@ -3,11 +3,11 @@ labels:
   recommended_framework: #the recommend framework, such as 'pytorch1.2.0'
   class: image-transform
   others:
-operator: 'towhee/transform-image-operator-class'
+operator: 'towhee/transform-image-operator-template'
 init:
   size: int
 call:
   input:
-    img: Union[np.ndarray, Image.Image, torch.Tensor, str]
+    img: Union[Image.Image, torch.Tensor, str]
   output:
     img_transformed: torch.Tensor
\ No newline at end of file