diff --git a/ee/backend/pkg/db/redis/client.go b/ee/backend/pkg/db/redis/client.go
index 52c4fd5f1..088bc8058 100644
--- a/ee/backend/pkg/db/redis/client.go
+++ b/ee/backend/pkg/db/redis/client.go
@@ -2,9 +2,11 @@ package redis
 
 import (
 	"errors"
-	"github.com/go-redis/redis"
-	config "openreplay/backend/internal/config/redis"
 	"strings"
+
+	"github.com/go-redis/redis"
+
+	config "openreplay/backend/internal/config/redis"
 )
 
 type Client struct {
diff --git a/ee/backend/pkg/db/redis/consumer.go b/ee/backend/pkg/db/redis/consumer.go
index d91cf4b88..eb89b48c1 100644
--- a/ee/backend/pkg/db/redis/consumer.go
+++ b/ee/backend/pkg/db/redis/consumer.go
@@ -3,15 +3,16 @@ package redis
 import (
 	"errors"
 	"fmt"
-	"github.com/go-redis/redis"
 	"log"
 	"net"
-	redis2 "openreplay/backend/pkg/db/redis"
-	"openreplay/backend/pkg/messages"
-	"openreplay/backend/pkg/queue/types"
 	"sort"
 	"strconv"
 	"strings"
+
+	"github.com/go-redis/redis"
+
+	"openreplay/backend/pkg/messages"
+	"openreplay/backend/pkg/queue/types"
 )
 
 type idsInfo struct {
@@ -21,7 +22,7 @@ type idsInfo struct {
 type streamPendingIDsMap map[string]*idsInfo
 
 type consumerImpl struct {
-	client     *redis2.Client
+	client     *Client
 	group      string
 	streams    []string
 	idsPending streamPendingIDsMap
@@ -45,7 +46,7 @@ func (c *consumerImpl) Close() {
 	panic("implement me")
 }
 
-func NewConsumer(client *redis2.Client, group string, streams []string) types.Consumer {
+func NewConsumer(client *Client, group string, streams []string) types.Consumer {
 	idsPending := make(streamPendingIDsMap)
 	streamsCount := len(streams)
 	for i := 0; i < streamsCount; i++ {
diff --git a/ee/backend/pkg/db/redis/producer.go b/ee/backend/pkg/db/redis/producer.go
index 091d572ce..adadb8163 100644
--- a/ee/backend/pkg/db/redis/producer.go
+++ b/ee/backend/pkg/db/redis/producer.go
@@ -1,21 +1,22 @@
 package redis
 
 import (
+	"log"
+
 	"github.com/go-redis/redis"
-	redis2 "openreplay/backend/pkg/db/redis"
+
 	"openreplay/backend/pkg/queue/types"
 )
 
 type producerImpl struct {
-	client *redis2.Client
+	client *Client
 }
 
 func (c *producerImpl) Close(timeout int) {
-	//TODO implement me
-	panic("implement me")
+	log.Printf("Redis producer close")
 }
 
-func NewProducer(client *redis2.Client) types.Producer {
+func NewProducer(client *Client) types.Producer {
 	return &producerImpl{
 		client: client,
 	}
diff --git a/ee/backend/pkg/sessions/redis.go b/ee/backend/pkg/sessions/redis.go
index 77dc72ce7..5f86dd53f 100644
--- a/ee/backend/pkg/sessions/redis.go
+++ b/ee/backend/pkg/sessions/redis.go
@@ -4,16 +4,16 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"openreplay/backend/pkg/db/redis"
-	"openreplay/backend/pkg/sessions"
 	"time"
+
+	"openreplay/backend/pkg/db/redis"
 )
 
 type cacheImpl struct {
 	db *redis.Client
 }
 
-func (c *cacheImpl) Set(session *sessions.Session) error {
+func (c *cacheImpl) Set(session *Session) error {
 	if c.db == nil {
 		return ErrDisabledCache
 	}
@@ -33,7 +33,7 @@ func (c *cacheImpl) Set(session *sessions.Session) error {
 	return nil
 }
 
-func (c *cacheImpl) Get(sessionID uint64) (*sessions.Session, error) {
+func (c *cacheImpl) Get(sessionID uint64) (*Session, error) {
 	if c.db == nil {
 		return nil, ErrDisabledCache
 	}
@@ -44,7 +44,7 @@ func (c *cacheImpl) Get(sessionID uint64) (*sessions.Session, error) {
 	if err != nil {
 		return nil, err
 	}
-	session := &sessions.Session{}
+	session := &Session{}
 	if err = json.Unmarshal([]byte(result), session); err != nil {
 		return nil, err
 	}
@@ -53,6 +53,6 @@ func (c *cacheImpl) Get(sessionID uint64) (*sessions.Session, error) {
 
 var ErrDisabledCache = errors.New("cache is disabled")
 
-func NewCache(db *redis.Client) sessions.Cache {
+func NewCache(db *redis.Client) Cache {
 	return &cacheImpl{db: db}
 }