diff --git a/nodes/generate_image_node_v2.py b/nodes/generate_image_node_v2.py
index 1baf513..929692f 100644
--- a/nodes/generate_image_node_v2.py
+++ b/nodes/generate_image_node_v2.py
@@ -22,9 +22,9 @@ class _BaseGenerateImageNodeV2:
                 "prompt": ("STRING",),
             },
             "optional": {
-                "mode": (["GAIA"], {"default": "GAIA"}),
+                "model_version": (["FIBO"], {"default": "FIBO"}),
                 "negative_prompt": ("STRING", {"default": ""}),
-                "image": ("IMAGE",),
+                "images": ("IMAGE",),
                 "aspect_ratio": (
                     ["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9"],
                     {"default": "1:1"},
@@ -47,17 +47,17 @@ class _BaseGenerateImageNodeV2:
     def _build_payload(
         self,
         prompt,
-        mode,
+        model_version,
         negative_prompt,
         aspect_ratio,
         steps_num,
         guidance_scale,
         seed,
-        image=None,
+        images=None,
     ):
         payload = {
             "prompt": prompt,
-            "mode": mode,
+            "model_version": model_version,
             "negative_prompt": negative_prompt,
             "aspect_ratio": aspect_ratio,
             "steps_num": steps_num,
@@ -65,10 +65,11 @@ class _BaseGenerateImageNodeV2:
             "seed": seed,
         }
 
-        if image is not None:
-            if isinstance(image, torch.Tensor):
-                image = preprocess_image(image)
-            payload["images"] = [image_to_base64(image)]
+        if images is not None:
+            if isinstance(images, torch.Tensor):
+                preprocess_images = preprocess_image(images)
+            payload["images"] = [image_to_base64(preprocess_images)]
+
 
         return payload
 
@@ -76,24 +77,24 @@ class _BaseGenerateImageNodeV2:
         self,
         api_token,
         prompt,
-        mode,
+        model_version,
         negative_prompt,
         aspect_ratio,
         steps_num,
         guidance_scale,
         seed,
-        image=None,
+        images=None,
     ):
         self._validate_token(api_token)
         payload = self._build_payload(
             prompt,
-            mode,
+            model_version,
             negative_prompt,
             aspect_ratio,
             steps_num,
             guidance_scale,
             seed,
-            image,
+            images,
         )
 
         headers = {"Content-Type": "application/json", "api_token": api_token}
diff --git a/nodes/refine_image_node_v2.py b/nodes/refine_image_node_v2.py
index ec30da3..ea6d71a 100644
--- a/nodes/refine_image_node_v2.py
+++ b/nodes/refine_image_node_v2.py
@@ -16,7 +16,7 @@ class _BaseRefineImageNodeV2:
                 "structured_prompt": ("STRING",),
             },
             "optional": {
-                "mode": (["GAIA"], {"default": "GAIA"}),
+                "model_version": (["FIBO"], {"default": "FIBO"}),
                 "negative_prompt": ("STRING", {"default": ""}),
                 "aspect_ratio": (
                     ["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9"],
@@ -41,7 +41,7 @@ class _BaseRefineImageNodeV2:
         self,
         prompt,
         structured_prompt,
-        mode,
+        model_version,
         negative_prompt,
         aspect_ratio,
         steps_num,
@@ -50,7 +50,7 @@ class _BaseRefineImageNodeV2:
     ):
         return {
             "prompt": prompt,
-            "mode": mode,
+            "model_version": model_version,
             "negative_prompt": negative_prompt,
             "aspect_ratio": aspect_ratio,
             "steps_num": steps_num,
@@ -64,7 +64,7 @@ class _BaseRefineImageNodeV2:
         api_token,
         prompt,
         structured_prompt,
-        mode,
+        model_version,
         negative_prompt,
         aspect_ratio,
         steps_num,
@@ -75,7 +75,7 @@ class _BaseRefineImageNodeV2:
         payload = self._build_payload(
             prompt,
             structured_prompt,
-            mode,
+            model_version,
             negative_prompt,
             aspect_ratio,
             steps_num,