diff --git a/api/graphql/endpoint/graphql_endpoint.go b/api/graphql/endpoint/graphql_endpoint.go
index a5edf7d..49b3e1f 100644
--- a/api/graphql/endpoint/graphql_endpoint.go
+++ b/api/graphql/endpoint/graphql_endpoint.go
@@ -9,6 +9,7 @@ import (
 	"github.com/99designs/gqlgen/graphql/handler/transport"
 	photoview_graphql "github.com/photoview/photoview/api/graphql"
 	"github.com/photoview/photoview/api/graphql/resolvers"
+	"github.com/photoview/photoview/api/server"
 	"github.com/photoview/photoview/api/utils"
 	"gorm.io/gorm"
 )
@@ -27,6 +28,7 @@ func GraphqlEndpoint(db *gorm.DB) *graphql_handler.Server {
 	graphqlServer := graphql_handler.New(photoview_graphql.NewExecutableSchema(graphqlConfig))
 	graphqlServer.AddTransport(transport.Websocket{
 		KeepAlivePingInterval: 10 * time.Second,
+		Upgrader:              server.WebsocketUpgrader(utils.DevelopmentMode()),
 	})
 	graphqlServer.AddTransport(transport.Options{})
 	graphqlServer.AddTransport(transport.GET{})