diff --git a/src/langtrace_python_sdk/instrumentation/ollama/patch.py b/src/langtrace_python_sdk/instrumentation/ollama/patch.py
index 5880abf0..ee29b031 100644
--- a/src/langtrace_python_sdk/instrumentation/ollama/patch.py
+++ b/src/langtrace_python_sdk/instrumentation/ollama/patch.py
@@ -84,7 +84,7 @@ async def traced_method(wrapped, instance, args, kwargs):
         try:
             result = await wrapped(*args, **kwargs)
             if kwargs.get("stream"):
-                return StreamWrapper(span, result)
+                return StreamWrapper(result, span)
             else:
                 _set_response_attributes(span, result)
             span.end()