diff --git a/management/client/client_test.go b/management/client/client_test.go
index 494b1e198..c7932da28 100644
--- a/management/client/client_test.go
+++ b/management/client/client_test.go
@@ -2,7 +2,7 @@ package client
 
 import (
 	"context"
-	"github.com/netbirdio/netbird/management/server/activity"
+	activity "github.com/netbirdio/netbird/management/server/activity/sqlite"
 	"net"
 	"path/filepath"
 	"sync"