diff --git a/rust/src_webpack/src/main.ts b/rust/src_webpack/src/main.ts
index 040a2e5..45e5e1e 100644
--- a/rust/src_webpack/src/main.ts
+++ b/rust/src_webpack/src/main.ts
@@ -125,8 +125,8 @@ export class Controller {
       controller.progressview.addProgress("ota_upload", (current / max) * 100, "Uploading firmeware (" + current + "/" + max + ")")
     }, false);
     ajax.addEventListener("load", () => {
-      //TODO wait for reboot here!
       controller.progressview.removeProgress("ota_upload")
+      controller.reboot();
     }, false);
     ajax.addEventListener("error", () => {
       alert("Error ota")