diff --git a/src/lib/ThemeToggle.svelte b/src/lib/ThemeToggle.svelte
index 5a2dcbf..ceed43c 100644
--- a/src/lib/ThemeToggle.svelte
+++ b/src/lib/ThemeToggle.svelte
@@ -1,9 +1,8 @@
 <script lang="ts">
 	import { afterUpdate } from "svelte";
 	import DarkMode from "svelte-dark-mode";
-	import type { Theme } from "svelte-dark-mode/types/DarkMode.svelte";
 
-	let theme: Theme;
+	let theme: "dark" | "light";
 
 	afterUpdate(() => {
 		document.documentElement.className = theme;
@@ -16,16 +15,4 @@
 
 <DarkMode bind:theme />
 
-{#if theme === "dark"}
-	<div on:click={toggle} class="i-fa6-solid:sun" />
-{:else if theme === "light"}
-	<div on:click={toggle} class="i-fa6-solid:moon" />
-{/if}
-
-<style>
-	div {
-		cursor: pointer;
-		display: flex;
-		align-items: center;
-	}
-</style>
+<div on:click={toggle} class="i-fa6-solid:{theme === "dark" ? "sun" : "moon"} cursor-pointer flex items-center" />
\ No newline at end of file
diff --git a/unocss.config.ts b/uno.config.ts
similarity index 87%
rename from unocss.config.ts
rename to uno.config.ts
index 0039a86..954df5f 100644
--- a/unocss.config.ts
+++ b/uno.config.ts
@@ -20,5 +20,7 @@ export default defineConfig({
 		})
 	],
 
-	transformers: [transformerVariantGroup(), transformerDirectives()]
+	transformers: [transformerVariantGroup(), transformerDirectives()],
+
+	safelist: ["i-fa6-solid:moon", "i-fa6-solid:sun"]
 });