diff --git a/pkg/armhelpers/graph.go b/pkg/armhelpers/graph.go
index ff664cd544..158668718a 100644
--- a/pkg/armhelpers/graph.go
+++ b/pkg/armhelpers/graph.go
@@ -42,7 +42,7 @@ func (az *AzureClient) DeleteRoleAssignmentByID(roleAssignmentID string) (author
 	return az.authorizationClient.DeleteByID(roleAssignmentID)
 }
 
-// List all role assignments for a principal (e.g. a VM) via the scope and the unique identifier of the principal
+// ListRoleAssignmentsForPrincipal (e.g. a VM) via the scope and the unique identifier of the principal
 func (az *AzureClient) ListRoleAssignmentsForPrincipal(scope string, principalID string) (authorization.RoleAssignmentListResult, error) {
 	filter := fmt.Sprintf("principalId eq '%s'", principalID)
 	return az.authorizationClient.ListForScope(scope, filter)
diff --git a/pkg/armhelpers/mockclients.go b/pkg/armhelpers/mockclients.go
index 332839fccd..047d184c7f 100644
--- a/pkg/armhelpers/mockclients.go
+++ b/pkg/armhelpers/mockclients.go
@@ -472,7 +472,7 @@ func (mc *MockACSEngineClient) DeleteRoleAssignmentByID(roleAssignmentID string)
 	return authorization.RoleAssignment{}, nil
 }
 
-// List all role assignments for a principal (e.g. a VM) via the scope and the unique identifier of the principal
+// ListRoleAssignmentsForPrincipal (e.g. a VM) via the scope and the unique identifier of the principal
 func (mc *MockACSEngineClient) ListRoleAssignmentsForPrincipal(scope string, principalID string) (authorization.RoleAssignmentListResult, error) {
 	roleAssignments := []authorization.RoleAssignment{}
 	return authorization.RoleAssignmentListResult{