1 files changed, 4 insertions, 5 deletions

diff --git a/cmd/deploy/deploy.go b/cmd/deploy/deploy.go
index 2b3ab4a..ee7ee4a 100644
--- a/cmd/deploy/deploy.go
+++ b/cmd/deploy/deploy.go
@@ -9,7 +9,6 @@ import (
         "strconv"
         "strings"
 
-        "github.com/bdw/deploy/internal/config"
         "github.com/bdw/deploy/internal/ssh"
         "github.com/bdw/deploy/internal/state"
         "github.com/bdw/deploy/internal/templates"
@@ -55,14 +54,14 @@ func runDeploy(args []string) {
 
         fs.Parse(args)
 
-        // Get host from flag or config
+        // Get host from flag or state default
         if *host == "" {
-                cfg, err := config.Load()
+                st, err := state.Load()
                 if err != nil {
-                        fmt.Fprintf(os.Stderr, "Error loading config: %v\n", err)
+                        fmt.Fprintf(os.Stderr, "Error loading state: %v\n", err)
                         os.Exit(1)
                 }
-                *host = cfg.Host
+                *host = st.GetDefaultHost()
         }
 
         if *host == "" || *domain == "" {