diff --git a/.azure/app-cloud-e2e.yml b/.azure/app-cloud-e2e.yml
index 15684983b3..ff62d9b9d0 100644
--- a/.azure/app-cloud-e2e.yml
+++ b/.azure/app-cloud-e2e.yml
@@ -169,14 +169,14 @@ jobs:
       condition: eq(variables['name'], 'template_react_ui')
       displayName: 'Clone Template React UI Repo'
 
-    - bash: pip install -q -r .actions/requirements.txt
-      displayName: 'Install assistant dependencies'
-
     # Fix imports to use `lightning` instead of `lightning_app` since we install lightning only ATM
     # TODO: fixme when installing `lightning_app` as well
     - bash: |
-        python .actions/assistant.py copy_replace_imports --source_dir="./examples" --source_import="lightning_app" --target_import="lightning.app"
-        python .actions/assistant.py copy_replace_imports --source_dir="./tests" --source_import="lightning_app" --target_import="lightning.app"
+        pip install -q -r .actions/requirements.txt
+        python .actions/assistant.py copy_replace_imports \
+          --source_dir="./examples" --source_import="lightning_app" --target_import="lightning.app"
+        python .actions/assistant.py copy_replace_imports \
+          --source_dir="./tests" --source_import="lightning_app" --target_import="lightning.app"
       displayName: 'Adjust examples'
 
     - bash: pip --version && pip list
diff --git a/.azure/app-flagships.yml b/.azure/app-flagships.yml
index 2c7b06dc74..a478fbe16e 100644
--- a/.azure/app-flagships.yml
+++ b/.azure/app-flagships.yml
@@ -41,7 +41,6 @@ jobs:
 
     # TODO:
     # - Training Studio
-    # - Muse
     # - Echo
     # - StreamLit / Gradio
     # - All homepage & docs apps
@@ -51,7 +50,10 @@ jobs:
         'App: Flashy':
           name: "flashy"
           repo: "https://github.com/Lightning-AI/LAI-Flashy-App.git"
-        'App: Jupyter':
+        'App: Muse':
+          name: "stable-diffusion"
+          repo: "https://github.com/Lightning-AI/stable-diffusion-deploy.git"
+        'Component: Jupyter':
           name: "jupyter"
           repo: "https://github.com/Lightning-AI/LAI-Jupyter-Component.git"
     timeoutInMinutes: "25"
@@ -72,14 +74,17 @@ jobs:
     - bash: |
         whoami
         mkdir -p tests/_flagships
-        mkdir -p $(video_artifact_dir)
+        mkdir -p $(video_artifact_dir)/$(name)
         printf "local id: $(local_id)\n"
         python --version
         pip --version
       displayName: 'Info'
 
-    - script: pip install -e .[cloud,test] -f https://download.pytorch.org/whl/cpu/torch_stable.html
-      displayName: 'Install Lightning & dependencies'
+    - script: |
+        pip install -q -r .actions/requirements.txt
+        python .actions/assistant.py copy_replace_imports \
+          --source_dir="./tests" --source_import="lightning_app" --target_import="lightning.app"
+      displayName: 'Adjust testing'
 
     - script: |
         pip install playwright
@@ -95,7 +100,14 @@ jobs:
         pip install -e .
       workingDirectory: tests/_flagships/$(name)
       condition: eq(variables['name'], 'flashy')
-      displayName: 'adjust env for Flashy'
+      displayName: 'adjust env -> Flashy'
+
+    - script: |
+        pip install -e _flagships/$(name)
+        cp _flagships/$(name)/tests/test_app.py integrations_app/flagship/test_stable-diffusion.py
+      workingDirectory: tests/
+      condition: eq(variables['name'], 'stable-diffusion')
+      displayName: 'adjust env -> Muse'
 
     - script: |
         pip install -e _flagships/$(name)
@@ -103,7 +115,10 @@ jobs:
         cp _flagships/$(name)/tests/test_jupyter_app.py integrations_app/flagship/test_jupyter.py
       workingDirectory: tests/
       condition: eq(variables['name'], 'jupyter')
-      displayName: 'adjust env for Jupyter'
+      displayName: 'adjust env -> Jupyter'
+
+    - script: pip install -e .[cloud,test] -f https://download.pytorch.org/whl/cpu/torch_stable.html
+      displayName: 'Install Lightning & dependencies'
 
     - bash: pip --version && pip list
       displayName: 'List pip dependency'
diff --git a/tests/integrations_app/flagship/test_stable-diffusion.py b/tests/integrations_app/flagship/test_stable-diffusion.py
new file mode 100644
index 0000000000..5d33698040
--- /dev/null
+++ b/tests/integrations_app/flagship/test_stable-diffusion.py
@@ -0,0 +1 @@
+# This is placeholder and the reals file ATM is being copied from the original repo/project