diff --git a/src/opengeodeweb_back/utils_functions.py b/src/opengeodeweb_back/utils_functions.py
index 2074026..934ac19 100644
--- a/src/opengeodeweb_back/utils_functions.py
+++ b/src/opengeodeweb_back/utils_functions.py
@@ -224,6 +224,7 @@ def generate_native_viewable_and_light_viewable_from_object(
 ) -> dict[str, str | list[str]]:
     data_entry = Data.create(
         geode_object=geode_object,
+        viewer_object=geode_functions.get_object_type(geode_object),
         input_file="",
         additional_files=[],
     )
@@ -236,6 +237,7 @@ def generate_native_viewable_and_light_viewable_from_file(
 ) -> dict[str, str | list[str]]:
     data_entry = Data.create(
         geode_object=geode_object,
+        viewer_object=geode_functions.get_object_type(geode_object),
         input_file=input_filename,
         additional_files=[],
     )
diff --git a/tests/test_models_routes.py b/tests/test_models_routes.py
index 83fa4cb..e1f916b 100644
--- a/tests/test_models_routes.py
+++ b/tests/test_models_routes.py
@@ -35,6 +35,7 @@ def test_extract_brep_uuids(client, test_id):
     with client.application.app_context():
         data_entry = Data.create(
             geode_object="BRep",
+            viewer_object=geode_functions.get_object_type("BRep"),
             input_file=brep_filename,
         )
         data_entry.native_file_name = brep_filename
diff --git a/tests/test_routes.py b/tests/test_routes.py
index fd1095e..17354b3 100644
--- a/tests/test_routes.py
+++ b/tests/test_routes.py
@@ -171,7 +171,11 @@ def get_full_data():
 
 def test_texture_coordinates(client, test_id):
     with client.application.app_context():
-        data = Data.create(geode_object="PolygonalSurface3D", input_file="hat.vtp")
+        data = Data.create(
+            geode_object="PolygonalSurface3D",
+            viewer_object=geode_functions.get_object_type("PolygonalSurface3D"),
+            input_file="hat.vtp",
+        )
         data.native_file_name = "hat.vtp"
         session = get_session()
         if session:
@@ -195,7 +199,11 @@ def test_vertex_attribute_names(client, test_id):
     route = f"/opengeodeweb_back/vertex_attribute_names"
 
     with client.application.app_context():
-        data = Data.create(geode_object="PolygonalSurface3D", input_file="test.vtp")
+        data = Data.create(
+            geode_object="PolygonalSurface3D",
+            viewer_object=geode_functions.get_object_type("PolygonalSurface3D"),
+            input_file="test.vtp",
+        )
         data.native_file_name = "test.vtp"
         session = get_session()
         if session:
@@ -217,7 +225,11 @@ def test_polygon_attribute_names(client, test_id):
     route = f"/opengeodeweb_back/polygon_attribute_names"
 
     with client.application.app_context():
-        data = Data.create(geode_object="PolygonalSurface3D", input_file="test.vtp")
+        data = Data.create(
+            geode_object="PolygonalSurface3D",
+            viewer_object=geode_functions.get_object_type("PolygonalSurface3D"),
+            input_file="test.vtp",
+        )
         data.native_file_name = "test.vtp"
         session = get_session()
         if session:
@@ -239,7 +251,11 @@ def test_polyhedron_attribute_names(client, test_id):
     route = f"/opengeodeweb_back/polyhedron_attribute_names"
 
     with client.application.app_context():
-        data = Data.create(geode_object="PolyhedralSolid3D", input_file="test.vtu")
+        data = Data.create(
+            geode_object="PolyhedralSolid3D",
+            viewer_object=geode_functions.get_object_type("PolyhedralSolid3D"),
+            input_file="test.vtu",
+        )
         data.native_file_name = "test.vtu"
         session = get_session()
         if session:
diff --git a/tests/test_utils_functions.py b/tests/test_utils_functions.py
index 3d7b3dd..87c457e 100644
--- a/tests/test_utils_functions.py
+++ b/tests/test_utils_functions.py
@@ -105,6 +105,7 @@ def test_save_all_viewables_and_return_info(client):
 
         data_entry = Data.create(
             geode_object=geode_object,
+            viewer_object=geode_functions.get_object_type(geode_object),
             input_file=input_file,
             additional_files=additional_files,
         )
@@ -146,6 +147,7 @@ def test_save_all_viewables_commits_to_db(client):
 
         data_entry = Data.create(
             geode_object=geode_object,
+            viewer_object=geode_functions.get_object_type(geode_object),
             input_file=input_file,
             additional_files=[],
         )