diff --git a/swarms/agents/utils/callback.py b/swarms/agents/utils/callback.py
index cca65037..771f370f 100644
--- a/swarms/agents/utils/callback.py
+++ b/swarms/agents/utils/callback.py
@@ -2,7 +2,7 @@ from typing import Any, Dict, List, Optional, Union
 
 from langchain.callbacks.base import BaseCallbackHandler
 from langchain.schema import AgentAction, AgentFinish, LLMResult
-# from celery import Task
+from celery import Task
 
 # from ansi import ANSI, Color, Style, dim_multiline
 from swarms.utils.main import ANSI, Color, Style, dim_multiline