diff --git a/main.go b/main.go
index db24d054c0a7f3c5117f0193a7dbcde912522b68..71288764ed2e2fd5da0ff9e8b50e20499848de3f 100644
--- a/main.go
+++ b/main.go
@@ -30,11 +30,13 @@ func main() {
 		log.Fatalln(err)
 	}
 	s, err := wish.NewServer(
-		fmt.Sprintf("%s:%d", cfg.Host, cfg.Port),
-		cfg.KeyPath,
-		bm.Middleware(tui.SessionHandler(cfg.RepoPath, time.Second*5)),
-		gm.Middleware(cfg.RepoPath, cfg.RepoAuth, cfg.RepoAuthFile),
-		lm.Middleware(),
+		wish.WithAddress(fmt.Sprintf("%s:%d", cfg.Host, cfg.Port)),
+		wish.WithHostKeyPath(cfg.KeyPath),
+		wish.WithMiddlewares(
+			bm.Middleware(tui.SessionHandler(cfg.RepoPath, time.Second*5)),
+			gm.Middleware(cfg.RepoPath, cfg.RepoAuth, cfg.RepoAuthFile),
+			lm.Middleware(),
+		),
 	)
 	if err != nil {
 		log.Fatalln(err)