diff --git a/src/rp_handler.py b/src/rp_handler.py
index 027b683..b2014ab 100644
--- a/src/rp_handler.py
+++ b/src/rp_handler.py
@@ -237,7 +237,7 @@ def process_output_images(outputs, job_id):
     for node_id, node_output in outputs.items():
         if "images" in node_output:
             for image in node_output["images"]:
-                output_images = image["filename"]
+                output_images = f"{image['subfolder']}/{image['filename']}"
 
     print(f"runpod-worker-comfy - image generation is done")
 
diff --git a/test_input.json b/test_input.json
index 1210cec..7fdb0f2 100644
--- a/test_input.json
+++ b/test_input.json
@@ -53,7 +53,7 @@
       },
       "9": {
         "inputs": {
-          "filename_prefix": "ComfyUI",
+          "filename_prefix": "ComfyUI/test",
           "images": ["8", 0]
         },
         "class_type": "SaveImage"