diff --git a/.azure-pipelines/generation-templates/generate_adapter-migrate-1es.yml b/.azure-pipelines/generation-templates/generate_adapter-migrate-1es.yml
index fd63e6559..dfa002b8a 100644
--- a/.azure-pipelines/generation-templates/generate_adapter-migrate-1es.yml
+++ b/.azure-pipelines/generation-templates/generate_adapter-migrate-1es.yml
@@ -6,6 +6,9 @@ parameters:
   - name: Sign
     type: boolean
     default: false
+  - name: Integration
+    type: boolean
+    default: false
 
 steps:
 - task: powershell@2
@@ -205,7 +208,34 @@ steps:
   inputs:
     testResultsFormat: NUnit
     testResultsFiles: "./test/results/pester-test-results-preview.xml"
-    failTaskOnFailedTests: true    
+    failTaskOnFailedTests: true
+- ${{ if eq(parameters.Integration, true) }}:
+  - task: powershell@2
+    displayName: 'Run Entra integration tests'
+    inputs:
+      targetType: inline
+      pwsh: true
+      script: |
+        cd test/module/Integration/Entra
+        Invoke-Pester -OutputFile "./test/results/pester-test-results-preview.xml" -OutputFormat NUnitXml
+  - task: PublishTestResults@2
+    inputs:
+      testResultsFormat: NUnit
+      testResultsFiles: "./test/results/pester-test-results-preview.xml"
+      failTaskOnFailedTests: true
+  - task: powershell@2
+    displayName: 'Run EntraBeta integration tests'
+    inputs:
+      targetType: inline
+      pwsh: true
+      script: |
+        cd test/module/Integration/EntraBeta
+        Invoke-Pester -OutputFile "./test/results/pester-test-results-preview.xml" -OutputFormat NUnitXml
+  - task: PublishTestResults@2
+    inputs:
+      testResultsFormat: NUnit
+      testResultsFiles: "./test/results/pester-test-results-preview.xml"
+      failTaskOnFailedTests: true
 - task: powershell@2
   displayName: 'Remove Local Gallery'
   inputs:
diff --git a/.azure-pipelines/generation-templates/generate_adapter.yml b/.azure-pipelines/generation-templates/generate_adapter.yml
index 52a8488c5..30567ca17 100644
--- a/.azure-pipelines/generation-templates/generate_adapter.yml
+++ b/.azure-pipelines/generation-templates/generate_adapter.yml
@@ -216,21 +216,20 @@ steps:
       targetType: inline
       pwsh: true
       script: |
-        cd test/module/Entra/Integration
+        cd test/module/Integration/Entra
         Invoke-Pester -OutputFile "./test/results/pester-test-results-preview.xml" -OutputFormat NUnitXml
   - task: PublishTestResults@2
     inputs:
       testResultsFormat: NUnit
       testResultsFiles: "./test/results/pester-test-results-preview.xml"
       failTaskOnFailedTests: true
-- ${{ if eq(parameters.Integration, true) }}:
   - task: powershell@2
     displayName: 'Run EntraBeta integration tests'
     inputs:
       targetType: inline
       pwsh: true
       script: |
-        cd test/module/EntraBeta/Integration
+        cd test/module/Integration/EntraBeta
         Invoke-Pester -OutputFile "./test/results/pester-test-results-preview.xml" -OutputFormat NUnitXml
   - task: PublishTestResults@2
     inputs: