diff --git a/pyproject.toml b/pyproject.toml
index d7e3631..e6c7a75 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "synth-sdk"
-version = "0.2.61"
+version = "0.2.64"
 description = ""
 authors = [{name = "Synth AI", email = "josh@usesynth.ai"}]
 license = {text = "MIT"}
diff --git a/setup.py b/setup.py
index fb277c7..68ce501 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="synth-sdk",
-    version="0.2.61",
+    version="0.2.64",
     packages=find_packages(),
     install_requires=[
         "opentelemetry-api",
diff --git a/synth_sdk/tracing/upload.py b/synth_sdk/tracing/upload.py
index 66f9189..a42b1b2 100644
--- a/synth_sdk/tracing/upload.py
+++ b/synth_sdk/tracing/upload.py
@@ -41,7 +41,7 @@ def send_system_traces(
 ):
     """Send all system traces and dataset metadata to the server."""
     # Get the token using the API key
-    token_url = f"{base_url}/token"
+    token_url = f"{base_url}/v1/auth/token"
     token_response = requests.get(
         token_url, headers={"customer_specific_api_key": api_key}
     )
@@ -50,7 +50,7 @@ def send_system_traces(
 
     # print("Traces: ", traces)
     # Send the traces with the token
-    api_url = f"{base_url}/upload/"
+    api_url = f"{base_url}/v1/uploads/"
 
     payload = createPayload(dataset, traces) # Create the payload
 
@@ -267,7 +267,7 @@ async def upload_helper(dataset: Dataset, traces: List[SystemTrace]=[], verbose:
         if show_payload:
             print("Payload sent to server: ")
             pprint(payload)
-        return response, payload
+        return response, payload, dataset, traces
     except ValueError as e:
         if verbose:
             print("Validation error:", str(e))