diff --git a/tests/data/__init__.py b/tests/data/__init__.py
index 7dcf49757..c93187222 100644
--- a/tests/data/__init__.py
+++ b/tests/data/__init__.py
@@ -19,13 +19,13 @@
 
 ### Constants used for data
 DATA_DIR = os.path.join(os.path.dirname(__file__))
-TWITTER_COMPLAINTS_DATA = os.path.join(DATA_DIR, "twitter_complaints_small.json")
+TWITTER_COMPLAINTS_DATA = os.path.join(DATA_DIR, "twitter_complaints_small.jsonl")
 TWITTER_COMPLAINTS_DATA_INPUT_OUTPUT = os.path.join(
-    DATA_DIR, "twitter_complaints_input_output.json"
+    DATA_DIR, "twitter_complaints_input_output.jsonl"
 )
 TWITTER_COMPLAINTS_JSON_FORMAT = os.path.join(DATA_DIR, "twitter_complaints_json.json")
 TWITTER_COMPLAINTS_TOKENIZED = os.path.join(
-    DATA_DIR, "twitter_complaints_tokenized_with_maykeye_tinyllama_v0.json"
+    DATA_DIR, "twitter_complaints_tokenized_with_maykeye_tinyllama_v0.jsonl"
 )
 EMPTY_DATA = os.path.join(DATA_DIR, "empty_data.json")
 MALFORMATTED_DATA = os.path.join(DATA_DIR, "malformatted_data.json")
diff --git a/tests/data/twitter_complaints_input_output.json b/tests/data/twitter_complaints_input_output.jsonl
similarity index 100%
rename from tests/data/twitter_complaints_input_output.json
rename to tests/data/twitter_complaints_input_output.jsonl
diff --git a/tests/data/twitter_complaints_small.json b/tests/data/twitter_complaints_small.jsonl
similarity index 100%
rename from tests/data/twitter_complaints_small.json
rename to tests/data/twitter_complaints_small.jsonl
diff --git a/tests/data/twitter_complaints_tokenized_with_maykeye_tinyllama_v0.json b/tests/data/twitter_complaints_tokenized_with_maykeye_tinyllama_v0.jsonl
similarity index 100%
rename from tests/data/twitter_complaints_tokenized_with_maykeye_tinyllama_v0.json
rename to tests/data/twitter_complaints_tokenized_with_maykeye_tinyllama_v0.jsonl