diff --git a/pyproject.toml b/pyproject.toml
index 65e1d7fb8..be833f673 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.5.39"
+version = "2.5.40"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
diff --git a/src/uipath/tracing/_utils.py b/src/uipath/tracing/_utils.py
index 4cdb66dea..4d8531715 100644
--- a/src/uipath/tracing/_utils.py
+++ b/src/uipath/tracing/_utils.py
@@ -88,7 +88,7 @@ class UiPathSpan:
     folder_key: Optional[str] = field(
         default_factory=lambda: env.get("UIPATH_FOLDER_KEY", "")
     )
-    source: Optional[str] = None
+    source: Optional[int] = None
     span_type: str = "Coded Agents"
     process_key: Optional[str] = field(
         default_factory=lambda: env.get("UIPATH_PROCESS_UUID")
@@ -294,6 +294,8 @@ def otel_span_to_uipath_span(
         # Top-level fields for internal tracing schema
         execution_type = attributes_dict.get("executionType")
         agent_version = attributes_dict.get("agentVersion")
+        reference_id = attributes_dict.get("referenceId")
+        source = attributes_dict.get("source")
 
         # Create UiPathSpan from OpenTelemetry span
         start_time = datetime.fromtimestamp(
@@ -322,6 +324,8 @@ def otel_span_to_uipath_span(
             span_type=span_type,
             execution_type=execution_type,
             agent_version=agent_version,
+            reference_id=reference_id,
+            source=source,
         )
 
     @staticmethod
diff --git a/uv.lock b/uv.lock
index e33500ba1..46a0dd5c6 100644
--- a/uv.lock
+++ b/uv.lock
@@ -2491,7 +2491,7 @@ wheels = [
 
 [[package]]
 name = "uipath"
-version = "2.5.39"
+version = "2.5.40"
 source = { editable = "." }
 dependencies = [
     { name = "applicationinsights" },