diff --git a/src/LMP.py b/src/LMP.py
index abecbc7..51e19ed 100644
--- a/src/LMP.py
+++ b/src/LMP.py
@@ -1,7 +1,6 @@
-
 import openai
 from time import sleep
-from openai.error import RateLimitError, APIConnectionError
+from openai import RateLimitError, APIConnectionError
 from pygments import highlight
 from pygments.lexers import PythonLexer
 from pygments.formatters import TerminalFormatter
@@ -76,7 +75,7 @@ def _cached_api_call(self, **kwargs):
                 print('(using cache)', end=' ')
                 return self._cache[kwargs]
             else:
-                ret = openai.ChatCompletion.create(**kwargs)['choices'][0]['message']['content']
+                ret = client.chat.completions.create(**kwargs).choices[0].message.content
                 # post processing
                 ret = ret.replace('```', '').replace('python', '').strip()
                 self._cache[kwargs] = ret
@@ -86,7 +85,7 @@ def _cached_api_call(self, **kwargs):
                 print('(using cache)', end=' ')
                 return self._cache[kwargs]
             else:
-                ret = openai.Completion.create(**kwargs)['choices'][0]['text'].strip()
+                ret = client.completions.create(**kwargs).choices[0].text.strip()
                 self._cache[kwargs] = ret
                 return ret
 
@@ -186,4 +185,4 @@ def exec_safe(code_str, gvars=None, lvars=None):
         exec(code_str, custom_gvars, lvars)
     except Exception as e:
         print(f'Error executing code:\n{code_str}')
-        raise e
\ No newline at end of file
+        raise e