diff --git a/src/stores/useModelStore.ts b/src/stores/useModelStore.ts
index c9ce865..a5ac3f5 100644
--- a/src/stores/useModelStore.ts
+++ b/src/stores/useModelStore.ts
@@ -100,6 +100,7 @@ interface ModelStore {
   setActiveModel: (id: string | null) => void;
   toggleModelVisible: (id: string) => void;
   toggleModelLocked: (id: string) => void;
+  setModelColor: (id: string, color: string) => void;
   duplicateModel: (id: string) => void;
   clearAllModels: () => void;
   maxModels: number;