diff --git a/ui/state.js b/ui/state.js
index fdd06fa..195f334 100644
--- a/ui/state.js
+++ b/ui/state.js
@@ -118,6 +118,7 @@ export function updateUI() {
       `input[name="amily2_lorebook_target"][value="${settings.lorebookTarget}"]`,
     ).prop("checked", true);
 
+    $(`input[name="amily2_icon_location"][value="${settings.iconLocation}"]`).prop("checked", true);
     populateModelDropdown(); 
   }
 }