diff --git a/src/stores/useModelStore.ts b/src/stores/useModelStore.ts
index 9d24072..0c81b9a 100644
--- a/src/stores/useModelStore.ts
+++ b/src/stores/useModelStore.ts
@@ -223,4 +223,7 @@ export const useModelStore = create<ModelStore>((set) => ({
     screenshots: state.screenshots.filter((_, i) => i !== index),
   })),
   clearScreenshots: () => set({ screenshots: [] }),
+
+  isLocked: false,
+  setLocked: (isLocked) => set({ isLocked }),
 }));