diff --git a/packages/adapters/sqlalchemy_impl/workspace_member_repository.py b/packages/adapters/sqlalchemy_impl/workspace_member_repository.py
index c96c42f86..74cf0fe90 100644
--- a/packages/adapters/sqlalchemy_impl/workspace_member_repository.py
+++ b/packages/adapters/sqlalchemy_impl/workspace_member_repository.py
@@ -2,13 +2,12 @@ from __future__ import annotations
 from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from sqlalchemy.orm import Session
-from packages.domain.entities import WorkspaceMember
 from packages.ports.workspace_member_repository import WorkspaceMemberRepository
 
 class SQLAlchemyWorkspaceMemberRepository(WorkspaceMemberRepository):
-    def __init__(self, session: Session):
+    def __init__(self, session=None):
         self.session = session
-    def save(self, member: WorkspaceMember) -> None:
+    def save(self, member) -> None:
         raise NotImplementedError("Workspace feature removed")
     def find_by_id(self, member_id: str):
         return None