diff --git a/packages/client/hmi-client/src/components/drilldown/tera-drilldown-section.vue b/packages/client/hmi-client/src/components/drilldown/tera-drilldown-section.vue
index ef8cd1df55..849044efd7 100644
--- a/packages/client/hmi-client/src/components/drilldown/tera-drilldown-section.vue
+++ b/packages/client/hmi-client/src/components/drilldown/tera-drilldown-section.vue
@@ -40,13 +40,13 @@ footer {
 header {
 	display: inline-flex;
 	justify-content: space-between;
-	padding: var(--gap-2) 0;
+	padding: var(--gap-4) 0;
 	gap: var(--gap-3);
 }
 
 header > div {
 	display: inline-flex;
-	gap: var(--gap-small);
+	gap: var(--gap-1);
 	align-items: center;
 }
 
diff --git a/packages/client/hmi-client/src/components/model/petrinet/model-configurations/tera-stratified-matrix-preview.vue b/packages/client/hmi-client/src/components/model/petrinet/model-configurations/tera-stratified-matrix-preview.vue
index 34a5844406..a6e0ae7693 100644
--- a/packages/client/hmi-client/src/components/model/petrinet/model-configurations/tera-stratified-matrix-preview.vue
+++ b/packages/client/hmi-client/src/components/model/petrinet/model-configurations/tera-stratified-matrix-preview.vue
@@ -18,12 +18,7 @@
 			</div>
 			<div class="matrix-grid">
 				<template v-for="row in matrix">
-					<div
-						class="cell"
-						:class="{ 'filled-cell': cell.content.id }"
-						v-for="(cell, colIdx) in row"
-						:key="colIdx"
-					></div>
+					<div class="cell" :class="{ 'filled-cell': cell.content.id }" v-for="(cell, colIdx) in row" :key="colIdx" />
 				</template>
 			</div>
 		</section>
diff --git a/packages/client/hmi-client/src/components/workflow/ops/model-from-equations/tera-model-from-equations-drilldown.vue b/packages/client/hmi-client/src/components/workflow/ops/model-from-equations/tera-model-from-equations-drilldown.vue
index 307854d64e..099e5a5f76 100644
--- a/packages/client/hmi-client/src/components/workflow/ops/model-from-equations/tera-model-from-equations-drilldown.vue
+++ b/packages/client/hmi-client/src/components/workflow/ops/model-from-equations/tera-model-from-equations-drilldown.vue
@@ -104,13 +104,7 @@
 				</template>
 			</tera-drilldown-section>
 			<tera-drilldown-preview>
-				<section v-if="selectedModel">
-					<tera-model-description
-						:model="selectedModel"
-						:feature-config="{ isPreview: true }"
-						:generating-card="isGeneratingCard"
-					/>
-				</section>
+				<tera-model-description v-if="selectedModel" :model="selectedModel" :generating-card="isGeneratingCard" />
 				<tera-operator-placeholder v-else :node="node" style="height: 100%" />
 			</tera-drilldown-preview>
 		</tera-columnar-panel>
@@ -120,7 +114,7 @@
 	</tera-drilldown>
 	<tera-save-asset-modal
 		v-if="selectedModel"
-		:model="selectedModel"
+		:asset="selectedModel"
 		:is-visible="showSaveModelModal"
 		@close-modal="onCloseModelModal"
 		@on-save="onAddModel"
diff --git a/packages/client/hmi-client/src/components/workflow/ops/stratify-mira/tera-stratify-mira.vue b/packages/client/hmi-client/src/components/workflow/ops/stratify-mira/tera-stratify-mira.vue
index 339fcf4d58..b920e887b0 100644
--- a/packages/client/hmi-client/src/components/workflow/ops/stratify-mira/tera-stratify-mira.vue
+++ b/packages/client/hmi-client/src/components/workflow/ops/stratify-mira/tera-stratify-mira.vue
@@ -10,41 +10,39 @@
 	>
 		<div :tabName="StratifyTabs.Wizard">
 			<tera-drilldown-section class="pl-4">
-				<div class="form-section">
-					<header class="inline-flex justify-content-between">
-						<section>
-							<h5>Stratify model</h5>
-							<p>The model will be stratified with the following settings.</p>
-							<p v-if="node.state.hasCodeBeenRun" class="code-executed-warning">
-								Note: Code has been executed which may not be reflected here.
-							</p>
-						</section>
-						<section>
-							<Button
-								style="margin-right: auto"
-								size="small"
-								severity="secondary"
-								outlined
-								label="Reset"
-								@click="resetModel"
-								class="mr-2"
-							/>
-							<Button
-								:disabled="isStratifyButtonDisabled"
-								:label="stratifyButtonLabel"
-								size="small"
-								icon="pi pi-play"
-								@click="stratifyModel"
-							/>
-						</section>
-					</header>
-					<tera-stratification-group-form
-						class="mt-2"
-						:model-node-options="modelNodeOptions"
-						:config="node.state.strataGroup"
-						@update-self="updateStratifyGroupForm"
+				<template #header-controls-left>
+					<section>
+						<h5>Stratify model</h5>
+						<p>The model will be stratified with the following settings.</p>
+						<p v-if="node.state.hasCodeBeenRun" class="code-executed-warning">
+							Note: Code has been executed which may not be reflected here.
+						</p>
+					</section>
+				</template>
+				<template #header-controls-right>
+					<Button
+						style="margin-right: auto"
+						size="small"
+						severity="secondary"
+						outlined
+						label="Reset"
+						@click="resetModel"
+						class="mr-2"
 					/>
-				</div>
+					<Button
+						:disabled="isStratifyButtonDisabled"
+						:label="stratifyButtonLabel"
+						size="small"
+						icon="pi pi-play"
+						@click="stratifyModel"
+					/>
+				</template>
+				<tera-stratification-group-form
+					class="mt-2"
+					:model-node-options="modelNodeOptions"
+					:config="node.state.strataGroup"
+					@update-self="updateStratifyGroupForm"
+				/>
 			</tera-drilldown-section>
 		</div>
 		<div :tabName="StratifyTabs.Notebook">
diff --git a/packages/client/hmi-client/src/model-representation/mira/mira-util.ts b/packages/client/hmi-client/src/model-representation/mira/mira-util.ts
index 826d466f9a..a8dd9af20b 100644
--- a/packages/client/hmi-client/src/model-representation/mira/mira-util.ts
+++ b/packages/client/hmi-client/src/model-representation/mira/mira-util.ts
@@ -169,7 +169,7 @@ export const extractTemplateMatrix = (templates: MiraTemplate[]) => {
 	const rowNames = extractConceptNames(templates, 'subject');
 	const colNames = extractConceptNames(templates, 'outcome');
 
-	// Not sure how to parse templates with no subejct or no outcomes
+	// Not sure how to parse templates with no subject or no outcomes
 	// and interacts with only controllers and itself, return a matrix
 	// with a single row for now - Mar 13, 2024
 	if (rowNames.length === 0 || colNames.length === 0) {
diff --git a/packages/client/hmi-client/src/model-representation/mira/mira.ts b/packages/client/hmi-client/src/model-representation/mira/mira.ts
index 28c131ad19..af15b87125 100644
--- a/packages/client/hmi-client/src/model-representation/mira/mira.ts
+++ b/packages/client/hmi-client/src/model-representation/mira/mira.ts
@@ -260,6 +260,7 @@ export const collapseTemplates = (miraModel: MiraModel) => {
 		if (!tempMatrixMap.has(key)) {
 			tempMatrixMap.set(key, []);
 		}
+
 		const originalTemplate = templateMap.get(t.name);
 		tempMatrixMap.get(key)?.push(originalTemplate as MiraTemplate);