diff --git a/constants.py b/constants.py
index cdb4758..d281694 100644
--- a/constants.py
+++ b/constants.py
@@ -1,4 +1,5 @@
-from typing import TypeAlias, TypedDict
+from typing_extensions import TypeAlias
+from typing import TypedDict
 
 FASTBOOT = 'fastboot'
 FLASH_PARTS = {
diff --git a/requirements.txt b/requirements.txt
index 47fc7b0..2b3138e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,3 +2,4 @@ click==8.0.1
 appdirs==1.4.4
 joblib==1.0.1
 toml
+typing_extensions