diff --git a/tools/walletextension/cache/RistrettoCache.go b/tools/walletextension/cache/RistrettoCache.go
index a36d7a3ed5..af417115b7 100644
--- a/tools/walletextension/cache/RistrettoCache.go
+++ b/tools/walletextension/cache/RistrettoCache.go
@@ -1,9 +1,10 @@
 package cache
 
 import (
-	"github.com/ethereum/go-ethereum/log"
 	"time"
 
+	"github.com/ethereum/go-ethereum/log"
+
 	"github.com/dgraph-io/ristretto"
 )
 
diff --git a/tools/walletextension/cache/cache.go b/tools/walletextension/cache/cache.go
index a46b2deb74..66e8b35f63 100644
--- a/tools/walletextension/cache/cache.go
+++ b/tools/walletextension/cache/cache.go
@@ -4,9 +4,10 @@ import (
 	"crypto/sha256"
 	"encoding/json"
 	"fmt"
-	"github.com/ethereum/go-ethereum/log"
 	"time"
 
+	"github.com/ethereum/go-ethereum/log"
+
 	"github.com/ten-protocol/go-ten/tools/walletextension/common"
 )
 
diff --git a/tools/walletextension/cache/cache_test.go b/tools/walletextension/cache/cache_test.go
index daecd66f82..f4bb05d941 100644
--- a/tools/walletextension/cache/cache_test.go
+++ b/tools/walletextension/cache/cache_test.go
@@ -5,6 +5,8 @@ import (
 	"testing"
 	"time"
 
+	"github.com/ethereum/go-ethereum/log"
+
 	"github.com/ten-protocol/go-ten/tools/walletextension/common"
 )
 
@@ -31,7 +33,8 @@ func TestGatewayCaching(t *testing.T) {
 	// cache tests
 	for name, test := range cacheTests {
 		t.Run(name, func(t *testing.T) {
-			cache, err := NewCache()
+			logger := log.New()
+			cache, err := NewCache(logger)
 			if err != nil {
 				t.Errorf("failed to create cache: %v", err)
 			}