diff --git a/pyproject.toml b/pyproject.toml
index 55bb0831c..1ec8e5b23 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langflow"
-version = "0.6.7a1"
+version = "0.6.7a2"
 description = "A Python package with a built-in web application"
 authors = ["Logspace <contact@logspace.ai>"]
 maintainers = [
diff --git a/src/backend/langflow/processing/base.py b/src/backend/langflow/processing/base.py
index 06dc0a063..fad53c254 100644
--- a/src/backend/langflow/processing/base.py
+++ b/src/backend/langflow/processing/base.py
@@ -2,10 +2,13 @@ from typing import TYPE_CHECKING, List, Union
 
 from langchain.agents.agent import AgentExecutor
 from langchain.callbacks.base import BaseCallbackHandler
+from loguru import logger
+
 from langflow.api.v1.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler
 from langflow.processing.process import fix_memory_inputs, format_actions
 from langflow.services.deps import get_plugins_service
-from loguru import logger
+from langflow.processing.process import fix_memory_inputs, format_actions
+from langflow.services.deps import get_plugins_service
 
 if TYPE_CHECKING:
     from langfuse.callback import CallbackHandler  # type: ignore
@@ -28,13 +31,12 @@ def setup_callbacks(sync, trace_id, **kwargs):
 
 def get_langfuse_callback(trace_id):
     from langflow.services.deps import get_plugins_service
-    from langfuse.callback import CreateTrace
 
     logger.debug("Initializing langfuse callback")
     if langfuse := get_plugins_service().get("langfuse"):
         logger.debug("Langfuse credentials found")
         try:
-            trace = langfuse.trace(CreateTrace(name="langflow-" + trace_id, id=trace_id))
+            trace = langfuse.trace(name="langflow-" + trace_id, id=trace_id)
             return trace.getNewHandler()
         except Exception as exc:
             logger.error(f"Error initializing langfuse callback: {exc}")
diff --git a/src/backend/langflow/services/plugins/langfuse_plugin.py b/src/backend/langflow/services/plugins/langfuse_plugin.py
index 424279342..e5d3afdd6 100644
--- a/src/backend/langflow/services/plugins/langfuse_plugin.py
+++ b/src/backend/langflow/services/plugins/langfuse_plugin.py
@@ -64,14 +64,13 @@ class LangfusePlugin(CallbackPlugin):
     def get_callback(self, _id: Optional[str] = None):
         if _id is None:
             _id = "default"
-        from langfuse.callback import CreateTrace  # type: ignore
 
         logger.debug("Initializing langfuse callback")
 
         try:
             langfuse_instance = self.get()
             if langfuse_instance is not None and hasattr(langfuse_instance, "trace"):
-                trace = langfuse_instance.trace(CreateTrace(name="langflow-" + _id, id=_id))
+                trace = langfuse_instance.trace(name="langflow-" + _id, id=_id)
                 if trace:
                     return trace.getNewHandler()