diff --git a/openai_examples/06_promptlayer/main.py b/openai_examples/06_promptlayer/main.py
index 72b9f10..fa9b36b 100644
--- a/openai_examples/06_promptlayer/main.py
+++ b/openai_examples/06_promptlayer/main.py
@@ -1,15 +1,15 @@
-import promptlayer
 import os
-promptlayer.api_key = os.environ.get('PROMPTLAYER_API_KEY')
+import promptlayer
 
-openai = promptlayer.openai
-openai.api_key = os.environ.get('OPENAI_API_KEY')
+promptlayer.api_key = os.environ.get('PROMPTLAYER_API_KEY')
+OpenAI = promptlayer.openai.OpenAI
+client = OpenAI()
 
 """
 Makes your request to OpenAI and then log the request to PromptLayer
 """
 
-response = openai.ChatCompletion.create(
+response = client.chat.completions.create(
     model='gpt-3.5-turbo',
     messages=[
         { 'role': 'system', 'content': 'You are a helpful assistant.' },