diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index ba9a0df..b73f5d4 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -59,6 +59,7 @@ jobs:
       - uses: actions/setup-go@v6
         with:
           go-version-file: kubeswarm-cli/go.mod
+          cache-dependency-path: kubeswarm-cli/go.sum
 
       - name: Build binary
         env:
diff --git a/cmd/swarm/main.go b/cmd/swarm/main.go
index 0548bc4..820f7a0 100644
--- a/cmd/swarm/main.go
+++ b/cmd/swarm/main.go
@@ -61,9 +61,9 @@ import (
 	"github.com/kubeswarm/kubeswarm/pkg/flow"
 	"github.com/kubeswarm/kubeswarm/pkg/observability"
 	"github.com/kubeswarm/kubeswarm/pkg/validation"
-	_ "github.com/kubeswarm/kubeswarm/runtime/providers/anthropic"
-	_ "github.com/kubeswarm/kubeswarm/runtime/providers/gemini"
-	_ "github.com/kubeswarm/kubeswarm/runtime/providers/openai"
+	_ "github.com/kubeswarm/kubeswarm/runtime/pkg/providers/anthropic"
+	_ "github.com/kubeswarm/kubeswarm/runtime/pkg/providers/gemini"
+	_ "github.com/kubeswarm/kubeswarm/runtime/pkg/providers/openai"
 )
 
 // version is set at build time via -ldflags "-X main.version=<tag>".