diff --git a/ts-editor/src/browser.ts b/ts-editor/src/browser.ts
index 57455e5..c392a28 100644
--- a/ts-editor/src/browser.ts
+++ b/ts-editor/src/browser.ts
@@ -191,6 +191,7 @@ function changeDisplayLayout() {
   W = w as number;
   H = h as number;
   ((window as any)._mdDriver as MonoDisplayDriver).setSize(W, H);
+  ((window as any)._mdDriver as MonoDisplayDriver).load(() => Promise.resolve(file.toBuffer()));
 }
 
 // --- Helpers -----------------------------------------------------------------
@@ -433,8 +434,7 @@ function triggerPreview() {
 function buildPreview() {
   if (!(window as any)._mdDriver)
     (window as any)._mdDriver = new MonoDisplayDriver("canvas_root", { onColor: "#EC0", offColor: "#000", fps: 25 });
-  ((window as any)._mdDriver as MonoDisplayDriver).load(() => Promise.resolve(file.toBuffer()));
-  ((window as any)._mdDriver as MonoDisplayDriver).setSize(W, H);
+  changeDisplayLayout();
 }
 
 // --- Load / Export -----------------------------------------------------------
diff --git a/ts/src/driver.ts b/ts/src/driver.ts
index e9004b7..fc56666 100644
--- a/ts/src/driver.ts
+++ b/ts/src/driver.ts
@@ -88,6 +88,11 @@ export class MonoDisplayDriver {
     }
   }
 
+  setSize(w: number, h: number) {
+    this.opts.displayHeight = h;
+    this.opts.displayWidth = w;
+  }
+
   stop(): void { this.renderer?.stop(); }
 }