diff --git a/tsdate/core.py b/tsdate/core.py
index fe80ca49..b1226f03 100644
--- a/tsdate/core.py
+++ b/tsdate/core.py
@@ -35,7 +35,7 @@
 import numpy as np
 import scipy.stats
 import tskit
-from tqdm import tqdm
+from tqdm.auto import tqdm
 
 from . import approx
 from . import base
diff --git a/tsdate/prior.py b/tsdate/prior.py
index c7a5965f..47df9c42 100644
--- a/tsdate/prior.py
+++ b/tsdate/prior.py
@@ -32,7 +32,7 @@
 import scipy.stats
 import tskit
 from scipy.special import comb
-from tqdm import tqdm
+from tqdm.auto import tqdm
 
 from . import base
 from . import cache