diff --git a/onethingai.py b/onethingai.py
index 73bbfed..f3333ef 100644
--- a/onethingai.py
+++ b/onethingai.py
@@ -26,7 +26,7 @@ class OneThingAILoader:
     
     RETURN_TYPES = ("IMAGE", )
     FUNCTION = "generate"
-    CATEGORY = "image generation"
+    CATEGORY = "OneThingAI/image generation"
 
     def generate(self, api_key, prompt, model="gpt4o", num_images=1, width=1024, height=1024, retries=3, timeout=8):
         # API endpoint
@@ -119,4 +119,4 @@ NODE_CLASS_MAPPINGS = {
 # Add descriptions
 NODE_DISPLAY_NAME_MAPPINGS = {
     "OneThingAILoader": "OneThingAI Image Generator"
-} 
\ No newline at end of file
+}