diff --git a/cmd/db/main.go b/cmd/db/main.go
index e17dd6f..898d4c7 100644
--- a/cmd/db/main.go
+++ b/cmd/db/main.go
@@ -5,7 +5,7 @@ import (
 	"os"
 	"strings"
 
-	"git.g3e.fr/syonad/two/internal/config"
+	configuration "git.g3e.fr/syonad/two/internal/config/agent"
 	"git.g3e.fr/syonad/two/pkg/db/kv"
 	"github.com/dgraph-io/badger/v4"
 )
@@ -125,19 +125,17 @@ func printDB() {
 }
 
 func main() {
-	configuration, err := config.LoadConfig("./two.yaml")
+	conf, err := configuration.LoadConfig("./two.yaml")
 	if err != nil {
 		fmt.Println(err)
 		return
 	}
 
 	DB = kv.InitDB(kv.Config{
-		Path: configuration.Database.Path,
+		Path: conf.Database.Path,
 	})
 	defer DB.Close()
 
-	printDB()
-
 	if len(os.Args) < 2 {
 		fmt.Println("Usage: db <cmd> [args...]")
 		return
@@ -186,6 +184,9 @@ func main() {
 		}
 		line, _ := GetFromDB(os.Args[2], os.Args[3])
 		fmt.Println(line)
+	case "print":
+		printDB()
+		os.Exit(1)
 	default:
 		fmt.Println("Unknown command:", cmd)
 		os.Exit(1)
diff --git a/internal/config/struct.go b/internal/config/agent/struct.go
similarity index 95%
rename from internal/config/struct.go
rename to internal/config/agent/struct.go
index a3708fb..b19d6da 100644
--- a/internal/config/struct.go
+++ b/internal/config/agent/struct.go
@@ -1,4 +1,4 @@
-package config
+package configuration
 
 import (
 	"github.com/spf13/viper"