diff --git a/src/command/Get-AzurePIMDirectoryRoles.ps1 b/src/command/Get-AzurePIMDirectoryRoles.ps1
index 5ceaa05..3881f39 100644
--- a/src/command/Get-AzurePIMDirectoryRoles.ps1
+++ b/src/command/Get-AzurePIMDirectoryRoles.ps1
@@ -33,7 +33,7 @@
             if ($skipAssignmentSettings) {
                 $_ | select *, @{n = 'PrincipalName'; e = { $_.principal.displayName } }, @{n = 'RoleName'; e = { $_.roleDefinition.displayName } }
             } else {
-                $rules = Get-PIMDirectoryRoleAssignmentSetting -roleId $_.roleDefinitionId -dontBeautify
+                $rules = Get-PIMDirectoryRoleAssignmentSetting $_.roleDefinition.templateId -dontBeautify
 
                 $_ | select *, @{n = 'PrincipalName'; e = { $_.principal.displayName } }, @{n = 'RoleName'; e = { $_.roleDefinition.displayName } }, @{n = 'Policy'; e = { $rules } }
             }
@@ -236,4 +236,4 @@
 
         $item | ConvertTo-Json -depth 100 | Out-File (New-Item -Path $outputFileName -Force)
     }
-}
\ No newline at end of file
+}