diff --git a/classes/DATASET_SaveImage.py b/classes/DATASET_SaveImage.py
index 1653bc4..80cf3ba 100644
--- a/classes/DATASET_SaveImage.py
+++ b/classes/DATASET_SaveImage.py
@@ -16,8 +16,8 @@ class DataSet_SaveImage:
         return {
             "required": {
                 "Images": ("IMAGE",),
+                "ImageFileNames": ("STRING", {"default": "Image"}),
                 "destination": ("STRING", {}),
-                "Filename": ("STRING", {"default": "Image"}),
             },
             "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
         }
@@ -27,7 +27,7 @@ class DataSet_SaveImage:
     OUTPUT_NODE = True
     CATEGORY = "🔶DATASET🔶"
 
-    def BatchSave(self, Images, destination, Filename, prompt=None, extra_pnginfo=None):
+    def BatchSave(self, Images, ImageFileNames, destination, prompt=None, extra_pnginfo=None):
 
         try:
 
diff --git a/classes/DataSet_TextFilesLoad.py b/classes/DataSet_TextFilesLoad.py
index 406e1cc..7c57fd3 100644
--- a/classes/DataSet_TextFilesLoad.py
+++ b/classes/DataSet_TextFilesLoad.py
@@ -26,7 +26,7 @@ class DataSet_TextFilesLoad:
         }
 
     INPUT_IS_LIST = True
-    RETURN_TYPES = ("STRING","STRING","STRING",)
+    RETURN_TYPES = ("STRING","STRING","STRING","STRING")
     RETURN_NAMES = ("TextFileNames","TextFileNamesWithoutExtension","TextFilePaths","TextFileContents",)
     OUTPUT_IS_LIST = (True, True, True, True,)
     FUNCTION = "LoadIT"