diff --git a/py/io/csv_prompts_loader.py b/py/io/csv_prompts_loader.py
index 188ab4e..06963ae 100644
--- a/py/io/csv_prompts_loader.py
+++ b/py/io/csv_prompts_loader.py
@@ -28,8 +28,10 @@ class CSVPromptsLoader():
             },
         }
 
-    RETURN_TYPES = ("STRING", "STRING", "STRING", "STRING", "STRING", )
-    RETURN_NAMES = ("styles", "face", "cloth", "action", "face_action",)
+    RETURN_TYPES = ("STRING", "STRING", "STRING",
+                    "STRING", "STRING", "STRING",)
+    RETURN_NAMES = ("styles", "face", "cloth", "action",
+                    "face_action", "environment",)
 
     FUNCTION = "process"
 
@@ -59,7 +61,7 @@ class CSVPromptsLoader():
         df = pd.read_csv(path, index_col=0, dtype={
                          "name": 'string', "styles": 'string', "face": 'string', "cloth": 'string', "action": 'string', "face_action": 'string', "environment": 'string'})
 
-        df.fillna("")
+        df = df.fillna("")
         if remove_extension == "On":
             key = os.path.splitext(key)[0]