diff --git a/common/tts_model.py b/common/tts_model.py
index f83a806..307c6c3 100644
--- a/common/tts_model.py
+++ b/common/tts_model.py
@@ -61,9 +61,9 @@ class Model:
     def get_style_vector_from_audio(
         self, audio_path: str, weight: float = 1.0
     ) -> np.ndarray:
-        from style_gen import extract_style_vector
+        from style_gen import get_style_vector
 
-        xvec = extract_style_vector(audio_path)
+        xvec = get_style_vector(audio_path)
         mean = self.style_vectors[0]
         xvec = mean + (xvec - mean) * weight
         return xvec