diff --git a/tests/e2e/Services/GraphQL/GraphQLBase.php b/tests/e2e/Services/GraphQL/GraphQLBase.php
index 95ad333c12..4d20f15751 100644
--- a/tests/e2e/Services/GraphQL/GraphQLBase.php
+++ b/tests/e2e/Services/GraphQL/GraphQLBase.php
@@ -993,14 +993,12 @@ trait GraphQLBase
                 }';
             case self::$GET_TEAM_MEMBERSHIP:
                 return 'query getTeamMembership($teamId: String!, $membershipId: String!){
-                    teamsGetMembership(teamId: $teamId, membershipId: $membershipId) {
-                        total
-                        memberships {
-                            _id
-                            teamId
-                            userName
-                            userEmail
-                        }
+                    teamsListMembership(teamId: $teamId, membershipId: $membershipId) {
+                        _id
+                        teamId
+                        userId
+                        userName
+                        userEmail
                     }
                 }';
             case self::$GET_TEAM_MEMBERSHIPS: