diff --git a/wshandler/wshandler.go b/wshandler/wshandler.go
index 77e1ffd..950da01 100644
--- a/wshandler/wshandler.go
+++ b/wshandler/wshandler.go
@@ -156,13 +156,13 @@ func init() {
 	gob.Register([]any{})
 }
 
-func NewWebsocketHandler(consumer session.Consumer, redisUri string) (*WebsocketHandler, error) {
+func NewWebsocketHandler(consumer session.Consumer) (*WebsocketHandler, error) {
 	var config wsConfig
 	if err := gocommon.LoadConfig(&config); err != nil {
 		return nil, err
 	}
 
-	redisSync, err := gocommon.NewRedisClient(redisUri)
+	redisSync, err := gocommon.NewRedisClient(config.Redis["wshandler"])
 	if err != nil {
 		return nil, err
 	}