diff --git a/src/lib/components/message.svelte b/src/lib/components/message.svelte
index 84cadba..8a78928 100644
--- a/src/lib/components/message.svelte
+++ b/src/lib/components/message.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-  import { type TypeMessage } from '../';
+  import { type TypeMessage } from '$lib';
   const { message, imageSrc, user }: TypeMessage = $props();
 </script>
 
diff --git a/src/lib/websocketConfig.ts b/src/lib/websocketConfig.ts
index 4c1553e..0da3a13 100644
--- a/src/lib/websocketConfig.ts
+++ b/src/lib/websocketConfig.ts
@@ -1,5 +1,7 @@
 import { Server as SocketIOServer } from 'socket.io';
 import type { HttpServer } from 'vite';
+// Don't try to replace with $lib alias. Since this 
+// file gets loaded as a vite plugin, it will crash
 import { client, createChannel, storeMessage } from './server/db/';
 import { v4 as uuidv4 } from 'uuid';
 
diff --git a/src/routes/+layout.svelte b/src/routes/+layout.svelte
index 34832d6..775c1a2 100644
--- a/src/routes/+layout.svelte
+++ b/src/routes/+layout.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-  import '../app.css';
+  import '$lib/app.css';
   let { children } = $props();
 </script>
 
diff --git a/src/routes/+page.server.ts b/src/routes/+page.server.ts
index 2a147dc..e87ecb7 100644
--- a/src/routes/+page.server.ts
+++ b/src/routes/+page.server.ts
@@ -1,6 +1,6 @@
 import type { PageLoad } from './$types';
 import type { TypeMessage } from '$lib';
-import { getMessages, client } from '../lib/server/db';
+import { getMessages, client } from '$/lib/server/db';
 
 export const load: PageLoad = async () => {
   const rows = await getMessages(client, '000', 5);
diff --git a/src/routes/+page.svelte b/src/routes/+page.svelte
index 9c6afc1..5d77c12 100644
--- a/src/routes/+page.svelte
+++ b/src/routes/+page.svelte
@@ -2,9 +2,9 @@
   import { io } from 'socket.io-client';
   import { onMount } from 'svelte';
   import { v4 as uuidv4 } from 'uuid';
-  import Message from '../lib/components/message.svelte';
-  import SendIcon from '../lib/icons/SendIcon.svelte';
-  import { type TypeMessage } from '../lib';
+  import Message from '$lib/components/message.svelte';
+  import SendIcon from '$lib/icons/SendIcon.svelte';
+  import { type TypeMessage } from '$lib';
   import type { PageData } from './$types';
 
   let { data }: { data: PageData } = $props();