diff --git a/ui/easydiffusion/easydb/mappings.py b/ui/easydiffusion/easydb/mappings.py
index 214a4c68..9a01f0c9 100644
--- a/ui/easydiffusion/easydb/mappings.py
+++ b/ui/easydiffusion/easydb/mappings.py
@@ -24,6 +24,7 @@ class GalleryImage(Base):
     use_upscale = Column(String)
     prompt = Column(String)
     negative_prompt = Column(String)
+    workspace = Column(String, server_default="default")
     time_created = Column(DateTime(timezone=True), server_default=func.now())
     nsfw = Column(Boolean, server_default=None)