diff --git a/tasks/eval_utils.py b/tasks/eval_utils.py
index 98d1bfb2ed..3078055f2d 100644
--- a/tasks/eval_utils.py
+++ b/tasks/eval_utils.py
@@ -11,7 +11,7 @@
 from megatron import get_args
 from megatron import print_rank_last, is_last_rank
 from megatron.core import mpu
-from megatron.schedules import get_forward_backward_func
+from megatron.core.pipeline_parallel import get_forward_backward_func
 from tasks.finetune_utils import build_data_loader
 from tasks.finetune_utils import process_batch