diff --git a/ee/api/chalicelib/core/sessions/__init__.py b/ee/api/chalicelib/core/sessions/__init__.py
index 4b9a40bc7..d03144e4f 100644
--- a/ee/api/chalicelib/core/sessions/__init__.py
+++ b/ee/api/chalicelib/core/sessions/__init__.py
@@ -6,13 +6,12 @@ logger = logging.getLogger(__name__)
 from . import sessions_pg
 from . import sessions_pg as sessions_legacy
 from . import sessions_ch
+from . import sessions_search as sessions_search_legacy
 
 if config("EXP_SESSIONS_SEARCH", cast=bool, default=False):
     logger.info(">>> Using experimental sessions search")
     from . import sessions_ch as sessions
-    from . import sessions_search as sessions_search_legacy
     from . import sessions_search_exp as sessions_search
 else:
     from . import sessions as sessions
     from . import sessions_search as sessions_search
-    from . import sessions_search as sessions_search_legacy
diff --git a/ee/api/chalicelib/core/sessions/sessions_search_exp.py b/ee/api/chalicelib/core/sessions/sessions_search_exp.py
index e53de4f6f..2c875bf57 100644
--- a/ee/api/chalicelib/core/sessions/sessions_search_exp.py
+++ b/ee/api/chalicelib/core/sessions/sessions_search_exp.py
@@ -3,7 +3,7 @@ import logging
 
 import schemas
 from chalicelib.core import metadata, projects
-from . import sessions_favorite, sessions as sessions_legacy, sessions_ch as sessions, sessions_legacy_mobil
+from . import sessions_favorite, sessions_search_legacy, sessions_ch as sessions, sessions_legacy_mobil
 from chalicelib.utils import pg_client, helper, ch_client, exp_ch_helper
 
 logger = logging.getLogger(__name__)
@@ -279,4 +279,4 @@ def search_by_metadata(tenant_id, user_id, m_key, m_value, project_id=None):
 # TODO: rewrite this function to use ClickHouse
 def search_sessions_by_ids(project_id: int, session_ids: list, sort_by: str = 'session_id',
                            ascending: bool = False) -> dict:
-    return sessions_legacy.search_sessions_by_ids(project_id, session_ids, sort_by, ascending)
+    return sessions_search_legacy.search_sessions_by_ids(project_id, session_ids, sort_by, ascending)