diff --git a/controller/update.go b/controller/update.go
index fec07b5..51f1f36 100644
--- a/controller/update.go
+++ b/controller/update.go
@@ -19,13 +19,17 @@ import (
 
 func UpdateApp(server *socketio.Server) <-chan string {
 	updates := make(chan string)
-	rootDir := filepath.Join("..")
+	exePath, _ := os.Executable()
+	exeDir := filepath.Dir(exePath)
+	rootDir := filepath.Join(exeDir, "..")
 
 	entries, err := os.ReadDir(rootDir)
 	if err != nil {
-		//log.Fatal("failed to read root dir: %v", err)
+		msg := fmt.Sprintf("failed to read root dir %s: %v", rootDir, err)
+		updates <- msg
+		server.BroadcastToRoom("/", "update", "updateLogs", msg)
 
-		server.BroadcastToRoom("/", "update", "updateLogs", fmt.Sprintf("failed to read root dir: %v", err))
+		return updates
 	}
 
 	var zips []string
@@ -130,6 +134,8 @@ func UpdateApp(server *socketio.Server) <-chan string {
 			server.BroadcastToRoom("/", "update", "updateLogs", msg)
 
 		}
+
+		updates <- "done"
 	}()
 
 	return updates