diff --git a/src/scripts/api.ts b/src/scripts/api.ts
index d46b4e031..9277f92f3 100644
--- a/src/scripts/api.ts
+++ b/src/scripts/api.ts
@@ -16,8 +16,8 @@ import type {
   HistoryTaskItem,
   LogsRawResponse,
   LogsWsMessage,
+  NodeProgressState,
   PendingTaskItem,
-  ProgressStateWsMessage,
   ProgressTextWsMessage,
   ProgressWsMessage,
   PromptResponse,
@@ -116,7 +116,7 @@ interface BackendApiCalls {
     promptId: string
   }
   progress_text: ProgressTextWsMessage
-  progress_state: ProgressStateWsMessage
+  progress_state: NodeProgressState
   display_component: DisplayComponentWsMessage
   feature_flags: FeatureFlagsWsMessage
 }