diff --git a/apps/notification/webapi/middleware/__init__.py b/apps/notification/webapi/middleware/__init__.py
index 68feba8..dfd330f 100644
--- a/apps/notification/webapi/middleware/__init__.py
+++ b/apps/notification/webapi/middleware/__init__.py
@@ -1,4 +1,4 @@
 from .freeleaps_auth_middleware import FreeleapsAuthMiddleware
-from .tenant_middleware import TenantMiddleware
+from .tenant_DB_connection_middleware import TenantDBConnectionMiddleware
 
-__all__ = ['FreeleapsAuthMiddleware', 'TenantMiddleware']
\ No newline at end of file
+__all__ = ['FreeleapsAuthMiddleware', 'TenantDBConnectionMiddleware']
\ No newline at end of file
diff --git a/apps/notification/webapi/providers/middleware.py b/apps/notification/webapi/providers/middleware.py
index 8eac8c1..0de606c 100644
--- a/apps/notification/webapi/providers/middleware.py
+++ b/apps/notification/webapi/providers/middleware.py
@@ -1,5 +1,5 @@
 from webapi.middleware.freeleaps_auth_middleware import FreeleapsAuthMiddleware
-from webapi.middleware.tenant_middleware import TenantMiddleware
+from webapi.middleware.tenant_DB_connection_middleware import TenantDBConnectionMiddleware
 
 
 def register(app):
@@ -7,5 +7,5 @@ def register(app):
     Register middleware to FastAPI application
     """
     # Register middlewares
-    app.add_middleware(TenantMiddleware)
+    app.add_middleware(TenantDBConnectionMiddleware)
     app.add_middleware(FreeleapsAuthMiddleware)
\ No newline at end of file