diff --git a/src/stores/useModelStore.ts b/src/stores/useModelStore.ts
index a5ac3f5..328ba54 100644
--- a/src/stores/useModelStore.ts
+++ b/src/stores/useModelStore.ts
@@ -244,6 +244,10 @@ export const useModelStore = create<ModelStore>((set, get) => ({
     models: state.models.map(m => m.id === id ? { ...m, locked: !m.locked } : m),
   })),
 
+  setModelColor: (id, color) => set((state) => ({
+    models: state.models.map(m => m.id === id ? { ...m, color } : m),
+  })),
+
   duplicateModel: (id) => {
     const state = get();
     if (state.models.length >= MAX_MODELS) return;