diff --git a/flask_mvp/auto_augmentation/progress.py b/flask_mvp/auto_augmentation/progress.py
index abe15fe35fb226fe94c30d513e5634b22a72c30a..5ddb04cf1c2094d6870e0793d78df59ebecfaedb 100644
--- a/flask_mvp/auto_augmentation/progress.py
+++ b/flask_mvp/auto_augmentation/progress.py
@@ -18,7 +18,7 @@ from tqdm import trange
 torch.manual_seed(0)
 # import agents and its functions
 
-from MetaAugment import UCB1_JC_py as UCB1_JC
+from MetaAugment.autoaugment_learners import ucb_learner as UCB1_JC
 
 from MetaAugment import Evo_learner as Evo