diff --git a/oneapi.py b/oneapi.py
index beb474a..14efb1f 100644
--- a/oneapi.py
+++ b/oneapi.py
@@ -139,7 +139,7 @@ async def _process_param_marker(node_data, var_spec, params):
     param_value = params[var_name]
     
     # Special handling for LoadImage node's image field
-    if node_data.get('class_type') == 'LoadImage' and input_field == 'image':
+    if node_data.get('class_type') == 'LoadImage':
         await _handle_load_image(node_data, param_value)
     else:
         # Regular parameter setting
diff --git a/workflows/example_img2img_workflow.json b/workflows/example_img2img_workflow.json
index 9468c1f..e27c8c1 100644
--- a/workflows/example_img2img_workflow.json
+++ b/workflows/example_img2img_workflow.json
@@ -99,7 +99,7 @@
     },
     "class_type": "LoadImage",
     "_meta": {
-      "title": "Load Image,$image.image"
+      "title": "Load Image,$image"
     }
   },
   "11": {