diff --git a/setup.cfg b/setup.cfg
index 19e6cda..e16fe36 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -53,6 +53,7 @@ install_requires =
     pymatgen
     matminer
     pybtex
+    typing-extensions
 
 
 [options.packages.find]
diff --git a/src/mp_time_split/core.py b/src/mp_time_split/core.py
index de18ecb..14fc19b 100644
--- a/src/mp_time_split/core.py
+++ b/src/mp_time_split/core.py
@@ -21,12 +21,13 @@
 from hashlib import md5
 from os import environ, path
 from pathlib import Path
-from typing import List, Literal, Optional, Tuple, Union
+from typing import List, Optional, Tuple, Union
 from urllib.request import urlretrieve
 
 import pandas as pd
 import pybtex.errors
 from matminer.utils.io import load_dataframe_from_json
+from typing_extensions import Literal
 
 from mp_time_split import __version__
 from mp_time_split.utils.data import DUMMY_SNAPSHOT_NAME, SNAPSHOT_NAME
diff --git a/src/mp_time_split/utils/api.py b/src/mp_time_split/utils/api.py
index c3796a1..1755a58 100644
--- a/src/mp_time_split/utils/api.py
+++ b/src/mp_time_split/utils/api.py
@@ -1,9 +1,10 @@
-from typing import List, Literal, Optional, Tuple, Union
+from typing import List, Optional, Tuple, Union
 
 import pandas as pd
 from mp_api import MPRester
 from mp_api.core.client import DEFAULT_API_KEY
 from tqdm import tqdm
+from typing_extensions import Literal
 
 from mp_time_split.utils.data import get_discovery_dict, noble, radioactive