1 files changed, 7 insertions, 6 deletions
diff --git a/main.go b/main.go
index dc955bc..4a78f3e 100644
--- a/main.go
+++ b/main.go
@@ -63,18 +63,19 @@ func main() {
 		os.Exit(2)
 	}
 
-	r := backend.NewRouter(dev, cfg)
+	assetsFS := backend.UsableEmbedFS("dist", embeds)
+	if dev {
+		assetsFS = os.DirFS("dist")
+	}
+
+	r := backend.NewRouter(dev, cfg, assetsFS)
 
 	backend.HandleHome(r)
 	backend.HandleRoot(r, cfg)
 	backend.HandleLogs(r)
 	backend.Handle404(r)
 
-	if dev {
-		backend.HandleStaticFiles(r, "/assets", os.DirFS("dist"))
-	} else {
-		backend.HandleStaticFiles(r, "/assets", backend.UsableEmbedFS("dist", embeds))
-	}
+	backend.HandleStaticFiles(r, "/assets", assetsFS)
 	backend.HandleStaticFiles(r, "/static", os.DirFS(cfg.PublicFolder))
 
 	slog.Info("starting http server")