From 3da48cd047dd246a75310e268af2791ec919e10c Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 15 Dec 2024 14:42:42 +0400
Subject: [PATCH] fix: single utt tests

---
 tests/unit/test_router.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index 067e78a9..1741865b 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -268,7 +268,7 @@ def test_add_single_utterance(
             index=index,
             auto_sync="local",
         )
-        route_layer.add(routes=[route_single_utterance])
+        route_layer.add(routes=route_single_utterance)
         assert route_layer.score_threshold == openai_encoder.score_threshold
         if index_cls is PineconeIndex:
             time.sleep(PINECONE_SLEEP)  # allow for index to be updated
@@ -286,7 +286,7 @@ def test_init_and_add_single_utterance(
         )
         if index_cls is PineconeIndex:
             time.sleep(PINECONE_SLEEP)  # allow for index to be updated
-        route_layer.add(routes=[route_single_utterance])
+        route_layer.add(routes=route_single_utterance)
         assert route_layer.score_threshold == openai_encoder.score_threshold
         _ = route_layer("Hello")
         assert len(route_layer.index.get_utterances()) == 1