diff --git a/test/db/test_entity_helper.py b/test/db/test_entity_helper.py
index e1b6d7be..b71cf96d 100644
--- a/test/db/test_entity_helper.py
+++ b/test/db/test_entity_helper.py
@@ -80,14 +80,13 @@ def test_valid_to_dict(self, datafile_entity):
                     "fileSize": 64,
                     "datafileModTime": str(Constants.TEST_MOD_CREATE_DATETIME),
                     "datafileCreateTime": str(Constants.TEST_MOD_CREATE_DATETIME),
-                    "dataset": 1,
                     "doi": "test doi",
                     "description": "test description",
                     "createId": "test create id",
                     "modId": "test mod id",
                     "datafileFormat": 1,
                     "createTime": str(Constants.TEST_MOD_CREATE_DATETIME),
-                    "DATASET": {
+                    "dataset": {
                         "id": None,
                         "createTime": None,
                         "modTime": None,
@@ -105,7 +104,7 @@ def test_valid_to_dict(self, datafile_entity):
                         "type": None,
                     },
                 },
-                "DATASET",
+                "dataset",
                 id="Dataset",
             ),
             pytest.param(
@@ -118,20 +117,18 @@ def test_valid_to_dict(self, datafile_entity):
                     "fileSize": 64,
                     "datafileModTime": str(Constants.TEST_MOD_CREATE_DATETIME),
                     "datafileCreateTime": str(Constants.TEST_MOD_CREATE_DATETIME),
-                    "dataset": 1,
                     "doi": "test doi",
                     "description": "test description",
                     "createId": "test create id",
                     "modId": "test mod id",
                     "datafileFormat": 1,
                     "createTime": str(Constants.TEST_MOD_CREATE_DATETIME),
-                    "DATASET": {
+                    "dataset": {
                         "id": None,
                         "createTime": None,
                         "modTime": None,
                         "createId": None,
                         "modId": None,
-                        "investigation": None,
                         "complete": None,
                         "description": None,
                         "doi": None,
@@ -141,7 +138,7 @@ def test_valid_to_dict(self, datafile_entity):
                         "startDate": None,
                         "sample": None,
                         "type": None,
-                        "INVESTIGATION": {
+                        "investigation": {
                             "id": None,
                             "createId": None,
                             "createTime": None,
@@ -160,7 +157,7 @@ def test_valid_to_dict(self, datafile_entity):
                         },
                     },
                 },
-                {"DATASET": "INVESTIGATION"},
+                {"dataset": "investigation"},
                 id="Dataset including investigation",
             ),
         ],