diff --git a/ATI/nodes.py b/ATI/nodes.py
index d679518..fee5a6f 100644
--- a/ATI/nodes.py
+++ b/ATI/nodes.py
@@ -105,6 +105,36 @@ def paint_point_track(
 
     return video
 
+def parse_json_tracks(tracks):
+    tracks_data = []
+    try:
+        # If tracks is a string, try to parse it as JSON
+        if isinstance(tracks, str):
+            parsed = json.loads(tracks.replace("'", '"'))
+            tracks_data.append(parsed)
+        else:
+            # If tracks is a list of strings, parse each one
+            for track_str in tracks:
+                parsed = json.loads(track_str.replace("'", '"'))
+                tracks_data.append(parsed)
+        
+        # Check if we have a single track (dict with x,y) or a list of tracks
+        if tracks_data and isinstance(tracks_data[0], dict) and 'x' in tracks_data[0]:
+            # Single track detected, wrap it in a list
+            tracks_data = [tracks_data]
+        elif tracks_data and isinstance(tracks_data[0], list) and tracks_data[0] and isinstance(tracks_data[0][0], dict) and 'x' in tracks_data[0][0]:
+            # Already a list of tracks, nothing to do
+            pass
+        else:
+            # Unexpected format
+            print(f"Warning: Unexpected track format: {type(tracks_data[0])}")
+            
+    except json.JSONDecodeError as e:
+        print(f"Error parsing tracks JSON: {e}")
+        tracks_data = []
+
+    return tracks_data
+
 class WanVideoATITracks:
     @classmethod
     def INPUT_TYPES(s):
@@ -126,33 +156,7 @@ class WanVideoATITracks:
     CATEGORY = "WanVideoWrapper"
 
     def patchmodel(self, model, tracks, width, height, temperature, topk, start_percent, end_percent):
-        tracks_data = []
-        try:
-            # If tracks is a string, try to parse it as JSON
-            if isinstance(tracks, str):
-                parsed = json.loads(tracks.replace("'", '"'))
-                tracks_data.append(parsed)
-            else:
-                # If tracks is a list of strings, parse each one
-                for track_str in tracks:
-                    parsed = json.loads(track_str.replace("'", '"'))
-                    tracks_data.append(parsed)
-            
-            # Check if we have a single track (dict with x,y) or a list of tracks
-            if tracks_data and isinstance(tracks_data[0], dict) and 'x' in tracks_data[0]:
-                # Single track detected, wrap it in a list
-                tracks_data = [tracks_data]
-            elif tracks_data and isinstance(tracks_data[0], list) and tracks_data[0] and isinstance(tracks_data[0][0], dict) and 'x' in tracks_data[0][0]:
-                # Already a list of tracks, nothing to do
-                pass
-            else:
-                # Unexpected format
-                print(f"Warning: Unexpected track format: {type(tracks_data[0])}")
-                
-        except json.JSONDecodeError as e:
-            print(f"Error parsing tracks JSON: {e}")
-            tracks_data = []
-
+        tracks_data = parse_json_tracks(tracks)
         arrs = []
         for track in tracks_data:
             pts = pad_pts(track)
@@ -189,18 +193,7 @@ class WanVideoATITracksVisualize:
     CATEGORY = "WanVideoWrapper"
 
     def patchmodel(self, images, tracks, min_radius, max_radius, max_retain):
-        tracks_data = []
-        if len(tracks) < 10:
-            for coords in tracks:
-                coords = json.loads(coords.replace("'", '"'))
-                tracks_data.append(coords)
-        else:
-            coords = json.loads(tracks.replace("'", '"'))
-            tracks_data.append(coords)
-
-        if tracks_data and isinstance(tracks_data[0], dict) and 'x' in tracks_data[0]:
-            tracks_data = [tracks_data]
-
+        tracks_data = parse_json_tracks(tracks)
         arrs = []
         for track in tracks_data:
             pts = pad_pts(track)