diff --git a/comfy/model_downloader_types.py b/comfy/model_downloader_types.py
index 2c5d522a2..788af7eb8 100644
--- a/comfy/model_downloader_types.py
+++ b/comfy/model_downloader_types.py
@@ -2,7 +2,8 @@ from __future__ import annotations
 
 import dataclasses
 from os.path import split
-from typing import Optional
+from typing import Optional, List
+from typing_extensions import TypedDict, NotRequired
 
 
 @dataclasses.dataclass
@@ -51,9 +52,6 @@ class HuggingFile:
         return split(self.filename)[-1]
 
 
-from typing import TypedDict, List, Optional, NotRequired
-
-
 class CivitStats(TypedDict):
     downloadCount: int
     favoriteCount: NotRequired[int]