diff --git a/main.py b/main.py
index 20193f82..3e25114d 100644
--- a/main.py
+++ b/main.py
@@ -291,6 +291,9 @@ class PromptServer(BaseHTTPRequestHandler):
                 info['output'] = obj_class.RETURN_TYPES
                 info['name'] = x #TODO
                 info['description'] = ''
+                info['category'] = 'sd'
+                if hasattr(obj_class, 'CATEGORY'):
+                    info['category'] = obj_class.CATEGORY
                 out[x] = info
             self.wfile.write(json.dumps(out).encode('utf-8'))
         elif self.path[1:] in os.listdir(self.server.server_dir):
diff --git a/webshit/index.html b/webshit/index.html
index 809517a4..31c36708 100644
--- a/webshit/index.html
+++ b/webshit/index.html
@@ -237,7 +237,7 @@ function onObjectInfo(json) {
         MyNode.__json_data = json[key]
 
         LiteGraph.registerNodeType(key, MyNode);
-        MyNode.category = "sd"; //TODO: proper categories
+        MyNode.category = json[key]['category'];
     };
 
     afterLoadGraph();