diff --git a/src/cortecs/fit/fit.py b/src/cortecs/fit/fit.py
index 9af9a09..b37d809 100644
--- a/src/cortecs/fit/fit.py
+++ b/src/cortecs/fit/fit.py
@@ -6,7 +6,7 @@
 from multiprocessing import Pool
 
 import numpy as np
-from tqdm.autonotebook import tqdm
+from tqdm.notebook import tqdm
 
 from cortecs.fit.fit_neural_net import *
 from cortecs.fit.fit_pca import *