diff --git a/atst/domain/workspace_roles.py b/atst/domain/workspace_roles.py
index f6942e2f..9da3224c 100644
--- a/atst/domain/workspace_roles.py
+++ b/atst/domain/workspace_roles.py
@@ -101,9 +101,9 @@ class WorkspaceRoles(object):
         return new_workspace_role
 
     @classmethod
-    def update_role(cls, member, workspace_id, role_name):
+    def update_role(cls, member, role_name):
         new_role = Roles.get(role_name)
-        workspace_role = WorkspaceRoles._get_workspace_role(member.user, workspace_id)
+        workspace_role = WorkspaceRoles._get_workspace_role(member.user, member.workspace_id)
         workspace_role.role = new_role
 
         db.session.add(workspace_role)
diff --git a/atst/domain/workspaces/workspaces.py b/atst/domain/workspaces/workspaces.py
index 453a1eed..6c43bc3b 100644
--- a/atst/domain/workspaces/workspaces.py
+++ b/atst/domain/workspaces/workspaces.py
@@ -119,7 +119,7 @@ class Workspaces(object):
             "edit workspace member",
         )
 
-        return WorkspaceRoles.update_role(member, workspace.id, role_name)
+        return WorkspaceRoles.update_role(member, role_name)
 
     @classmethod
     def _create_workspace_role(