diff --git a/src/index.d.ts b/src/index.d.ts
index 0209a72..64498f9 100644
--- a/src/index.d.ts
+++ b/src/index.d.ts
@@ -1,8 +1,8 @@
 export function init(options?: {
-  toggle?: HTMLElement;
+  toggle?: Element | null;
   classes?: { dark: string, light: string };
   default?: string;
   storageKey?: string;
-  onInit?: (toggle?: HTMLElement) => unknown;
-  onChange?: (theme?: string, toggle?: HTMLElement) => unknown;
+  onInit?: (toggle?: Element | null) => unknown;
+  onChange?: (theme?: string, toggle?: Element | null) => unknown;
 }): void;