diff --git a/src/text_generation/chat_gpt.py b/src/text_generation/text_generator.py
similarity index 100%
rename from src/text_generation/chat_gpt.py
rename to src/text_generation/text_generator.py
diff --git a/tests/text_generation/test_chat_gpt.py b/tests/text_generation/test_text_generator.py
similarity index 50%
rename from tests/text_generation/test_chat_gpt.py
rename to tests/text_generation/test_text_generator.py
index 0f9884f..413121f 100644
--- a/tests/text_generation/test_chat_gpt.py
+++ b/tests/text_generation/test_text_generator.py
@@ -1,8 +1,8 @@
 import pytest
-import src.text_generation.chat_gpt as chat_gpt
+import src.text_generation.text_generator as text_generator
 
 @pytest.mark.apitest
 def test_chat_gpt_response():
     prompt = "Say anything!"
-    response = chat_gpt.chat_with_gpt(prompt)
+    response = text_generator.chat_with_gpt(prompt)
     assert response
\ No newline at end of file