1 files changed, 3 insertions, 3 deletions

diff --git a/relay_test.go b/relay_test.go
index b39aa06..38b2062 100644
--- a/relay_test.go
+++ b/relay_test.go
@@ -9,13 +9,13 @@ import (
         "testing"
         "time"
 
-        "github.com/coder/websocket"
+        "northwest.io/nostr/internal/websocket"
 )
 
 // mockRelay creates a test WebSocket server that echoes messages
 func mockRelay(t *testing.T, handler func(conn *websocket.Conn)) *httptest.Server {
         return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-                conn, err := websocket.Accept(w, r, nil)
+                conn, err := websocket.Accept(w, r)
                 if err != nil {
                         t.Logf("Failed to accept WebSocket: %v", err)
                         return
@@ -77,7 +77,7 @@ func TestRelaySendReceive(t *testing.T) {
         ctx := context.Background()
 
         // Create relay without auto-Listen to test Send/Receive directly
-        conn, _, err := websocket.Dial(ctx, url, nil)
+        conn, err := websocket.Dial(ctx, url)
         if err != nil {
                 t.Fatalf("Dial() error = %v", err)
         }