diff --git a/langfuse/_client/client.py b/langfuse/_client/client.py
index df243e51c..df0173303 100644
--- a/langfuse/_client/client.py
+++ b/langfuse/_client/client.py
@@ -257,7 +257,7 @@ def __init__(
             secret_key=secret_key,
             host=self._host,
             timeout=timeout,
-            environment=environment,
+            environment=self._environment,
             release=release,
             flush_at=flush_at,
             flush_interval=flush_interval,
diff --git a/tests/test_openai.py b/tests/test_openai.py
index 85205db28..623802e55 100644
--- a/tests/test_openai.py
+++ b/tests/test_openai.py
@@ -94,6 +94,7 @@ def test_openai_chat_completion_stream(openai):
     assert len(chat_content) > 0
 
     langfuse.flush()
+    sleep(1)
 
     generation = get_api().observations.get_many(
         name=generation_name, type="GENERATION"