diff --git a/ComfyI2I.py b/ComfyI2I.py
index 89eaddf..23f8cba 100644
--- a/ComfyI2I.py
+++ b/ComfyI2I.py
@@ -657,7 +657,7 @@ class Mask_Ops:
             inputs = CLIPSegProcessor.from_pretrained(model, cache_dir=cache)
             model = CLIPSegForImageSegmentation.from_pretrained(model, cache_dir=cache)
             
-            image = tensor2pil(image)
+            image = image.convert('RGB')
             cache = os.path.join(MODELS_DIR, 'clipseg')
 
             with torch.no_grad():
@@ -1010,4 +1010,4 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "Mask Ops": "Mask Ops",
     "Inpaint Segments": "Inpaint Segments",
     "Combine and Paste": "Combine and Paste",
-}
\ No newline at end of file
+}