diff --git a/py/lama.py b/py/lama.py
index 03b265d..4d3f8d7 100644
--- a/py/lama.py
+++ b/py/lama.py
@@ -67,7 +67,7 @@ class LaMa:
                     _mask = tensor2pil(expand_mask(image2mask(_mask), mask_grow, mask_blur))
                 
                 ret_image = pixel_spread(tensor2pil(_image).convert('RGB'), ImageChops.invert(_mask.convert('RGB')))
-                ret_images.append(ret_image)
+                ret_images.append(pil2tensor(ret_image))
         else:
             temp_dir = os.path.join(folder_paths.get_temp_directory(), generate_random_name('_lama_', '_temp', 16))
             if os.path.isdir(temp_dir):
@@ -122,4 +122,4 @@ NODE_CLASS_MAPPINGS = {
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "LayerUtility: LaMa": "LayerUtility: LaMa"
-}
\ No newline at end of file
+}