diff --git a/packages/svelte/src/lib/container/SvelteFlow/types.ts b/packages/svelte/src/lib/container/SvelteFlow/types.ts
index 9ec9a37d..9d4513aa 100644
--- a/packages/svelte/src/lib/container/SvelteFlow/types.ts
+++ b/packages/svelte/src/lib/container/SvelteFlow/types.ts
@@ -18,7 +18,6 @@ import type {
   OnConnect,
   OnConnectStart,
   OnConnectEnd,
-  ColorModeClass,
   OnReconnect,
   OnRecoonnectStart,
   OnReconnectEnd
@@ -353,7 +352,7 @@ export type SvelteFlowProps<
      */
     colorMode?: ColorMode;
     /** Fallback color mode for SSR if colorMode is set to 'system' */
-    colorModeSSR?: ColorModeClass;
+    colorModeSSR?: Omit<ColorMode, 'system'>;
     /** Class to be applied to the flow container */
     class?: ClassValue;
     /** Styles to be applied to the flow container */
diff --git a/packages/svelte/src/lib/plugins/Controls/types.ts b/packages/svelte/src/lib/plugins/Controls/types.ts
index 7b011a22..c9ddfa6e 100644
--- a/packages/svelte/src/lib/plugins/Controls/types.ts
+++ b/packages/svelte/src/lib/plugins/Controls/types.ts
@@ -1,5 +1,5 @@
 import type { Snippet } from 'svelte';
-import type { ClassValue, HTMLAttributes, HTMLButtonAttributes } from 'svelte/elements';
+import type { ClassValue, HTMLAttributes } from 'svelte/elements';
 import type { PanelPosition } from '@xyflow/system';
 
 import type { FitViewOptions } from '$lib/types';
@@ -31,7 +31,7 @@ export type ControlsProps = {
   fitViewOptions?: FitViewOptions;
 } & HTMLAttributes<HTMLDivElement>;
 
-export type ControlButtonProps = HTMLButtonAttributes & {
+export type ControlButtonProps = HTMLAttributes<HTMLButtonElement> & {
   class?: ClassValue;
   bgColor?: string;
   bgColorHover?: string;