diff --git a/WebServer/Components/Pages/Home.razor b/WebServer/Components/Pages/Home.razor
index 940bde83..e5f72b2b 100644
--- a/WebServer/Components/Pages/Home.razor
+++ b/WebServer/Components/Pages/Home.razor
@@ -193,10 +193,9 @@
             Task thread = new Task(async () => {
                 aiModule.TrainAI();
                 trainButtonText = "AI Trained";
-                StateHasChanged();
                 await Task.Delay(2000);
                 trainButtonText = "Train AI";
-                StateHasChanged();
+                await Task.Delay(1);
                 Debounce = true;
             });
             thread.Start();