From ff7e5e8f323ec69479974b9a99d5c5493e0f9406 Mon Sep 17 00:00:00 2001
From: luboslenco <lubos.lenco@gmail.com>
Date: Sat, 6 Sep 2025 21:17:28 +0200
Subject: [PATCH] Fix saving project icon

---
 paint/sources/ui_menubar.ts | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/paint/sources/ui_menubar.ts b/paint/sources/ui_menubar.ts
index 4dd93446..5a0bea7b 100644
--- a/paint/sources/ui_menubar.ts
+++ b/paint/sources/ui_menubar.ts
@@ -55,16 +55,14 @@ function ui_menubar_render_ui() {
 
 			if (ui_menubar_icon_button(0, 2)) box_preferences_show();
 			if (ui_menubar_icon_button(0, 3)) {
-				///if (arm_android || arm_ios)
 				// Save project icon in lit mode
-				if (context_raw.viewport_mode == viewport_mode_t.PATH_TRACE) {
-					context_raw.viewport_mode = viewport_mode_t.LIT;
-				}
+				context_set_viewport_mode(viewport_mode_t.LIT);
 				console_toast(tr("Saving project"));
-				project_save();
-				///end
 				sys_notify_on_next_frame(function () {
-					box_projects_show();
+					sys_notify_on_end_frame(function () {
+						project_save();
+						box_projects_show();
+					});
 				});
 			}
 			if (ui_menubar_icon_button(4, 2)) {