diff --git a/src/main.zig b/src/main.zig
index 7ff7eed..c95c735 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -29,8 +29,7 @@ fn onGui() void {
     defer pm_gui.end();
 
     ui.beginFrame();
+    defer ui.endFrame();
 
     editor.update();
-
-    ui.endFrame();
 }