diff --git a/src/frontend/src/types/api/index.ts b/src/frontend/src/types/api/index.ts
index 16c840e37..46e62376c 100644
--- a/src/frontend/src/types/api/index.ts
+++ b/src/frontend/src/types/api/index.ts
@@ -17,6 +17,7 @@ export type APIClassType = {
   description: string;
   template: APITemplateType;
   display_name: string;
+  icon?: string;
   input_types?: Array<string>;
   output_types?: Array<string>;
   custom_fields?: CustomFieldsType;