diff --git a/agents-api/agents_api/activities/task_steps/prompt_step.py b/agents-api/agents_api/activities/task_steps/prompt_step.py
index 0b67a2aa9..8da0e49c7 100644
--- a/agents-api/agents_api/activities/task_steps/prompt_step.py
+++ b/agents-api/agents_api/activities/task_steps/prompt_step.py
@@ -17,6 +17,7 @@
 from ...common.utils.template import render_template
 from ..utils import get_handler_with_filtered_params
 from .base_evaluate import base_evaluate
+from ...env import debug
 
 COMPUTER_USE_BETA_FLAG = "computer-use-2024-10-22"