diff --git a/pages/3_🔍_Search.py b/pages/3_🔍_Search.py
index c0a7b84..a9abad5 100644
--- a/pages/3_🔍_Search.py
+++ b/pages/3_🔍_Search.py
@@ -1,7 +1,8 @@
 import streamlit as st
 
+from open_notebook.domain.models import model_manager
 from open_notebook.domain.notebook import text_search, vector_search
-from open_notebook.models import EmbeddingModel, model_manager
+from open_notebook.models import EmbeddingModel
 from pages.stream_app.note import note_list_item
 from pages.stream_app.source import source_list_item
 from pages.stream_app.utils import setup_page
diff --git a/pages/stream_app/utils.py b/pages/stream_app/utils.py
index 69bf78e..88f8849 100644
--- a/pages/stream_app/utils.py
+++ b/pages/stream_app/utils.py
@@ -2,8 +2,8 @@ import streamlit as st
 from loguru import logger
 
 from open_notebook.database.migrate import MigrationManager
+from open_notebook.domain.models import model_manager
 from open_notebook.graphs.chat import ThreadState, graph
-from open_notebook.models import model_manager
 from open_notebook.utils import (
     compare_versions,
     get_installed_version,