diff --git a/src/opengeodeweb_microservice/database/data.py b/src/opengeodeweb_microservice/database/data.py
index 8605bb5..7e65640 100644
--- a/src/opengeodeweb_microservice/database/data.py
+++ b/src/opengeodeweb_microservice/database/data.py
@@ -13,6 +13,7 @@ class Data(Base):
     )
     geode_object: Mapped[str] = mapped_column(String, nullable=False)
     viewer_object: Mapped[str] = mapped_column(String, nullable=False)
+    viewer_elements_type: Mapped[str] = mapped_column(String, nullable=False)
 
     native_file: Mapped[str | None] = mapped_column(String, nullable=True)
     viewable_file: Mapped[str | None] = mapped_column(String, nullable=True)
@@ -25,12 +26,14 @@ class Data(Base):
     def create(
         geode_object: str,
         viewer_object: str,
+        viewer_elements_type: str,
         input_file: str | None = None,
         additional_files: list[str] | None = None,
     ) -> "Data":
         data_entry = Data(
             geode_object=geode_object,
             viewer_object=viewer_object,
+            viewer_elements_type=viewer_elements_type,
             input_file=input_file,
             additional_files=additional_files,
         )
diff --git a/tests/test_database.py b/tests/test_database.py
index 1105d8d..05ce69c 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -5,6 +5,7 @@ def test_data_crud_operations(clean_database: None) -> None:
     data = Data.create(
         geode_object="test_object",
         viewer_object="test_viewer",
+        viewer_elements_type="test_type",
         input_file="test.txt",
         additional_files=[],
     )
@@ -27,6 +28,7 @@ def test_data_with_additional_files(clean_database: None) -> None:
     data = Data.create(
         geode_object="test_files",
         viewer_object="test_viewer",
+        viewer_elements_type="test_type",
         additional_files=files,
     )
     assert data.id is not None