diff --git a/nodes/flex/feature_extractors_whisper.py b/nodes/flex/feature_extractors_whisper.py
index c327590..e15e2be 100644
--- a/nodes/flex/feature_extractors_whisper.py
+++ b/nodes/flex/feature_extractors_whisper.py
@@ -355,7 +355,7 @@ class WhisperToPromptTravel:
     def INPUT_TYPES(cls):
         return {
             "required": {
-                "segments_alignment": ("STRING", {"multiline": True}), # JSON string of segment alignments
+                "alignment_data": ("whisper_alignment",),
                 "fps": ("FLOAT", {"default": 24.0, "min": 0.1, "max": 120.0}),
             }
         }