1 files changed, 3 insertions, 4 deletions

diff --git a/testing/forefront_test.py b/testing/forefront_test.py
index 4d5f4bc1..092d1a74 100644
--- a/testing/forefront_test.py
+++ b/testing/forefront_test.py
@@ -1,10 +1,9 @@
-import forefront
+from openai_rev import forefront
 
 # create an account
 token = forefront.Account.create(logging=True)
 print(token)
 
 # get a response
-for response in forefront.StreamingCompletion.create(token=token,
-                                                     prompt='hello world', model='gpt-4'):
-    print(response.completion.choices[0].text, end='')
+for response in forefront.StreamingCompletion.create(token=token, prompt='hello world', model='gpt-4'):
+    print(response)