diff --git a/image-generator-demo/src/app/app.element.scss b/image-generator-demo/src/app/app.element.scss
index 4473358..b056665 100644
--- a/image-generator-demo/src/app/app.element.scss
+++ b/image-generator-demo/src/app/app.element.scss
@@ -5,6 +5,10 @@
 	background: #242433;
 }
 
+sts-scene-data-provider {
+	display: contents;
+}
+
 // Layout utilities
 .v-start, .v-end, .h-start, .h-end, .h-center {
 	position: absolute;
diff --git a/image-generator-demo/src/app/app.element.ts b/image-generator-demo/src/app/app.element.ts
index 3ea93a9..af2aaa5 100644
--- a/image-generator-demo/src/app/app.element.ts
+++ b/image-generator-demo/src/app/app.element.ts
@@ -18,6 +18,7 @@ import { ImageInfo, LcmClient } from "./lcm-client";
 
 import "@storyteller/studio-web/canvas-provider";
 import "@storyteller/studio-web/loading";
+// import "@storyteller/studio-web/scene-data-provider";
 import "@storyteller/studio-web/scene-hierarchy";
 import "@storyteller/studio-web/studio";
 import "@storyteller/studio-web/transform-toolbar";
@@ -173,6 +174,7 @@ export class AppElement extends LitElement {
 
 	protected override render = () => html`
 		<sts-canvas-provider class="app-root">
+		<sts-scene-data-provider>
 			<div
 				class="hierarchy-wrapper"
 				style=${styleMap({ width: `${this.hierarchyWidth}px` })}
@@ -297,6 +299,7 @@ export class AppElement extends LitElement {
 			${this.loading ? html`
 				<sts-loading></sts-loading>
 			` : nothing}
+		</sts-scene-data-provider>
 		</sts-canvas-provider>
 	`
 }