diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index cbf10f10fd9e90aab86a06fe7c7f300c49291726..ba612d640abb3841bc3d28cdcf19ff1854999c1c 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -1,6 +1,6 @@
 import os
 import tempfile
-from unittest.mock import mock_open, patch, MagicMock, create_autospec
+from unittest.mock import mock_open, patch
 
 import pytest
 
@@ -8,7 +8,6 @@ from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder
 from semantic_router.layer import LayerConfig, RouteLayer
 from semantic_router.route import Route
 from semantic_router.llms.base import BaseLLM
-from semantic_router.schema import Message
 
 
 def mock_encoder_call(utterances):
@@ -370,9 +369,6 @@ class TestRouteLayer:
             ]
         }"""
 
-        # Instead of mocking, directly instantiate BaseLLM with a name
-        fake_llm_instance = BaseLLM(name="fake-model-v1")
-
         config_path = tmp_path / "config_with_llm.json"
         with open(config_path, "w") as file:
             file.write(llm_config_json)