diff --git a/system/tts_engines/f5tts/model_engine.py b/system/tts_engines/f5tts/model_engine.py
index 896af53..88dab71 100644
--- a/system/tts_engines/f5tts/model_engine.py
+++ b/system/tts_engines/f5tts/model_engine.py
@@ -79,7 +79,7 @@ def install_and_restart():
         raise ImportError("Could not install required packages")
 
 try:
-    from f5_tts.model import CFM, DiT
+    from f5_tts.model import CFM, DiT, UNetT
     from f5_tts.model.utils import (
         get_tokenizer,
         convert_char_to_pinyin,
@@ -1102,4 +1102,4 @@ class tts_class:
             except Exception as e:
                 print(f"[{self.branding}Debug] Device verification warning: {str(e)}") if self.debug_tts else None
             
-            self.tts_generating_lock = False
\ No newline at end of file
+            self.tts_generating_lock = False