diff --git a/packages/design-system/src/components/N8nButton/index.d.ts b/packages/design-system/src/components/N8nButton/index.d.ts
index 62fb0e1feb..b70b0b8418 100644
--- a/packages/design-system/src/components/N8nButton/index.d.ts
+++ b/packages/design-system/src/components/N8nButton/index.d.ts
@@ -3,6 +3,9 @@ import { N8nComponent, N8nComponentSize } from '../component';
 /** Button type */
 export type ButtonType = 'primary' | 'outline' | 'light' | 'text';
 
+/** Button themes */
+export type ButtonTheme = 'success' | 'warning' | 'danger';
+
 /** Button Component */
 export declare class N8nButton extends N8nComponent {
 	/** Button text */
@@ -12,7 +15,7 @@ export declare class N8nButton extends N8nComponent {
 	title: string;
 
 	/** Color scheme */
-	theme: string;
+	theme: ButtonTheme;
 
 	/** Button size */
 	size: N8nComponentSize;
diff --git a/packages/design-system/src/components/N8nIconButton/index.d.ts b/packages/design-system/src/components/N8nIconButton/index.d.ts
index 1de4cd7777..9a30ed5d05 100644
--- a/packages/design-system/src/components/N8nIconButton/index.d.ts
+++ b/packages/design-system/src/components/N8nIconButton/index.d.ts
@@ -1,7 +1,11 @@
 import { N8nComponent, N8nComponentSize } from '../component';
+import { ButtonTheme, ButtonType } from '../N8nButton';
 
 /** Icon Button Component */
 export declare class N8nIconButton extends N8nComponent {
+	/** Button type */
+	type: ButtonType;
+
 	/** Button title on hover */
 	title: string;
 
@@ -19,4 +23,7 @@ export declare class N8nIconButton extends N8nComponent {
 
 	/** Button icon, accepts an icon name of font awesome icon component */
 	icon: string;
+
+	/** Button theme */
+	theme: ButtonTheme;
 }