diff --git a/tests/test_rag_engine.py b/tests/test_rag_engine.py
index b5a1f34..34f28f7 100644
--- a/tests/test_rag_engine.py
+++ b/tests/test_rag_engine.py
@@ -31,7 +31,8 @@ def test_rag_engine_chroma(mock_chroma, mock_embed):
  vs_config = models.VectorStoreConfig(provider='chroma', collection_name='test')
  config = models.RAGConfig(enabled=True, vector_store=vs_config, embedding_provider='local')
  
- with patch('src.rag_engine.SentenceTransformer') as mock_st:
+ with patch('src.rag_engine._get_sentence_transformers') as mock_st:
+  mock_st.return_value = MagicMock()
   engine = RAGEngine(config)
   assert engine.collection == mock_collection