diff --git a/src/sockets.tsx b/src/sockets.tsx
index 62446af..78a65d8 100644
--- a/src/sockets.tsx
+++ b/src/sockets.tsx
@@ -29,7 +29,9 @@ const onError = (error: Event) => {
 
 export const connect = () => {
   if (isNotConnected()) {
-    const cacheServer = localStorage.getItem('cacheServer') ??
+    const cacheServer =
+      localStorage.getItem('cacheServer') ||
+      import.meta.env.PRIMAL_CACHE_URL ||
       'wss://cache3.primal.net/cache17';
 
     setSocket(new WebSocket(cacheServer));
diff --git a/src/uploadSocket.tsx b/src/uploadSocket.tsx
index bb16f54..df8d704 100644
--- a/src/uploadSocket.tsx
+++ b/src/uploadSocket.tsx
@@ -29,7 +29,8 @@ const onError = (error: Event) => {
 
 export const connect = () => {
   if (isNotConnected()) {
-    const cacheServer = localStorage.getItem('uploadServer') ??
+    const cacheServer = localStorage.getItem('uploadServer') ||
+      import.meta.env.PRIMAL_UPLOAD_URL ||
       'wss://uploads.primal.net/v1';
 
     setSocket(new WebSocket(cacheServer));
diff --git a/vite.config.ts b/vite.config.ts
index 476f5ba..b8a8b81 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -10,4 +10,5 @@ export default defineConfig({
     target: 'esnext',
     sourcemap: true,
   },
+  envPrefix: 'PRIMAL_',
 });