diff --git a/tests/unit/encoders/test_tfidf.py b/tests/unit/encoders/test_tfidf.py
index 7664433d070d74d06c00b7e0fb0336fa6fb7bc6e..5052a035360a513e7513d37882a21ac76cb8bd9c 100644
--- a/tests/unit/encoders/test_tfidf.py
+++ b/tests/unit/encoders/test_tfidf.py
@@ -37,7 +37,7 @@ class TestTfidfEncoder:
         result = tfidf_encoder(["test"])
         assert isinstance(result, list), "Result should be a list"
         assert all(
-            isinstance(sublist, list) for sublist in result
+            isinstance(sparse_emb.embedding, np.ndarray) for sparse_emb in result
         ), "Each item in result should be a list"
 
     def test_call_method_no_docs_tfidf_encoder(self, tfidf_encoder):
@@ -55,7 +55,7 @@ class TestTfidfEncoder:
         result = tfidf_encoder(["doc with fake word gta5jabcxyz"])
         assert isinstance(result, list), "Result should be a list"
         assert all(
-            isinstance(sublist, list) for sublist in result
+            isinstance(sparse_emb.embedding, np.ndarray) for sparse_emb in result
         ), "Each item in result should be a list"
 
     def test_call_method_with_uninitialized_model(self, tfidf_encoder):
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index ef36e0ab4b4180b3ed5e5746d984691efd2b5150..a20c68e1c541bfc875e0bbb7e4fe4cfd9c74b0ec 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -19,6 +19,7 @@ from platform import python_version
 PINECONE_SLEEP = 6
 
 
+
 def mock_encoder_call(utterances):
     # Define a mapping of utterances to return values
     mock_responses = {