diff --git a/nesta_ds_utils/viz/altair/saving.py b/nesta_ds_utils/viz/altair/saving.py
index 86c060c..1fe48f1 100644
--- a/nesta_ds_utils/viz/altair/saving.py
+++ b/nesta_ds_utils/viz/altair/saving.py
@@ -3,7 +3,7 @@
 """
 
 import altair_saver as alt_saver
-from altair.vegalite.v4.api import Chart
+from altair.vegalite import Chart
 import altair as alt
 from selenium import webdriver
 from webdriver_manager.chrome import ChromeDriverManager