diff --git a/src/server.ts b/src/server.ts
index 3eb7a6c..26742a2 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -195,6 +195,7 @@ export function createServer(port: number): WebSocketServer {
 }
 
 if (isMain) {
-  createServer(3001);
-  console.log("DiscoveryServer listening on ws://localhost:3001");
+  const port = parseInt(process.env.PORT || "3001", 10);
+  createServer(port);
+  console.log(`DiscoveryServer listening on port ${port}`);
 }