Make docker build work

1 files changed, 7 insertions, 1 deletions

diff --git a/main.go b/main.go
index 7cc3e90..a9c084d 100644
--- a/main.go
+++ b/main.go
@@ -4,6 +4,7 @@ package main
 
 import (
         "embed"
+        "io/fs"
         "log"
 
         "code.crute.us/mcrute/ssh-proxy/app"
@@ -35,7 +36,12 @@ func main() {
         }
         cli.AddFlags(rootCmd, &app.Config{}, app.DefaultConfig, "")
 
-        web.Register(rootCmd, embeddedTemplates, embeddedClients, appVersion)
+        templates, err := fs.Sub(embeddedTemplates, "templates")
+        if err != nil {
+                log.Fatalf("Error building sub-fs of embeded fs")
+        }
+
+        web.Register(rootCmd, templates, embeddedClients, appVersion)
         client.Register(rootCmd)
         register.Register(rootCmd)