diff --git a/tests/test_rag_integration.py b/tests/test_rag_integration.py
index 27eb24c..5ddaa24 100644
--- a/tests/test_rag_integration.py
+++ b/tests/test_rag_integration.py
@@ -32,7 +32,7 @@ def test_rag_integration(mock_project):
  with patch('src.app_controller.AppController._fetch_models'), \
       patch('src.models.load_config', return_value={}), \
       patch('src.paths.get_full_path_info', return_value={'logs_dir': {'path': mock_project}, 'scripts_dir': {'path': mock_project}}), \
-      patch('src.theme.load_from_config'):
+      patch('src.theme_2.load_from_config'):
   app = AppController()
   # Minimal state setup for _handle_request_event
   app.ui_global_system_prompt = ""