diff --git a/nodes/tasks.py b/nodes/tasks.py
index 4782973..a663e42 100644
--- a/nodes/tasks.py
+++ b/nodes/tasks.py
@@ -101,7 +101,7 @@ class gtUIPromptImageGenerationTask(gtUIBaseTask):
 
     RETURN_TYPES = ("IMAGE", "STRING")
     RETURN_NAMES = ("IMAGE", "file_path")
-    CATEGORY = "Griptape/Preview"
+    CATEGORY = "Griptape/Images"
 
     def run(
         self,