diff --git a/code/aind_auto_train/curriculums/coupled_baiting_2p1.py b/code/aind_auto_train/curriculums/coupled_baiting_2p1.py
index 0f4f45d..338fb3f 100644
--- a/code/aind_auto_train/curriculums/coupled_baiting_2p1.py
+++ b/code/aind_auto_train/curriculums/coupled_baiting_2p1.py
@@ -17,6 +17,8 @@
     Task, TrainingStage, DynamicForagingParas, 
     AutoWaterMode, AdvancedBlockMode
 )
+from aind_auto_train import setup_logging
+setup_logging()
 
 curriculum_name = Task.C1B1
 curriculum_version = "2.1"
diff --git a/code/aind_auto_train/curriculums/uncoupled_baiting_2p1.py b/code/aind_auto_train/curriculums/uncoupled_baiting_2p1.py
index 54b4a17..c8a7b00 100644
--- a/code/aind_auto_train/curriculums/uncoupled_baiting_2p1.py
+++ b/code/aind_auto_train/curriculums/uncoupled_baiting_2p1.py
@@ -19,6 +19,8 @@
     Task, TrainingStage, DynamicForagingParas,
     AutoWaterMode, AdvancedBlockMode
 )
+from aind_auto_train import setup_logging
+setup_logging()
 
 # Note this could be any string, not necessarily one of the Task enums
 curriculum_name = "Uncoupled Baiting"
diff --git a/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p1.py b/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p1.py
index cf49b91..815c127 100644
--- a/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p1.py
+++ b/code/aind_auto_train/curriculums/uncoupled_no_baiting_2p1.py
@@ -18,6 +18,8 @@
     Task, TrainingStage, DynamicForagingParas,
     AutoWaterMode, AdvancedBlockMode
 )
+from aind_auto_train import setup_logging
+setup_logging()
 
 # Note this could be any string, not necessarily one of the Task enums
 curriculum_name = Task.C0B0