diff --git a/src/llmtuner/extras/packages.py b/src/llmtuner/extras/packages.py
index 2e60e690..22cab732 100644
--- a/src/llmtuner/extras/packages.py
+++ b/src/llmtuner/extras/packages.py
@@ -19,7 +19,7 @@ _jieba_available = is_package_available("jieba")
 _matplotlib_available = is_package_available("matplotlib")
 _nltk_available = is_package_available("nltk")
 _rouge_available = is_package_available("rouge_chinese")
-_starlette_available = is_package_available("sse-starlette")
+_starlette_available = is_package_available("sse_starlette")
 _uvicorn_available = is_package_available("uvicorn")