diff --git a/plugins/host_websocket.go b/plugins/host_websocket.go
index f9152c061..40fab8445 100644
--- a/plugins/host_websocket.go
+++ b/plugins/host_websocket.go
@@ -59,11 +59,11 @@ type webSocketServiceImpl struct {
 }
 
 // newWebSocketService creates a new WebSocketService for a plugin.
-func newWebSocketService(pluginName string, manager *Manager, allowedHosts []string) *webSocketServiceImpl {
+func newWebSocketService(pluginName string, manager *Manager, permission *WebSocketPermission) *webSocketServiceImpl {
 	return &webSocketServiceImpl{
 		pluginName:   pluginName,
 		manager:      manager,
-		allowedHosts: allowedHosts,
+		allowedHosts: permission.AllowedHosts,
 		connections:  make(map[string]*wsConnection),
 	}
 }
diff --git a/plugins/manager_loader.go b/plugins/manager_loader.go
index 3d9aa99fb..1aca9fdc8 100644
--- a/plugins/manager_loader.go
+++ b/plugins/manager_loader.go
@@ -63,7 +63,7 @@ var hostServices = []hostServiceEntry{
 		registerStubs: func() []extism.HostFunction { return host.RegisterWebSocketHostFunctions(nil) },
 		create: func(ctx *serviceContext) ([]extism.HostFunction, io.Closer) {
 			perm := ctx.permissions.Websocket
-			service := newWebSocketService(ctx.pluginName, ctx.manager, perm.AllowedHosts)
+			service := newWebSocketService(ctx.pluginName, ctx.manager, perm)
 			return host.RegisterWebSocketHostFunctions(service), service
 		},
 	},