diff --git a/llama_cpp/llama_cpp.py b/llama_cpp/llama_cpp.py
index 1939190..3da131e 100644
--- a/llama_cpp/llama_cpp.py
+++ b/llama_cpp/llama_cpp.py
@@ -15,9 +15,9 @@ from typing import (
     Optional,
     TYPE_CHECKING,
     TypeVar,
-    TypeAlias,
     Generic,
 )
+from typing_extensions import TypeAlias
 
 
 # Load the library