diff --git a/test/datagateway_api/db/endpoints/test_create_db.py b/test/datagateway_api/db/endpoints/test_create_db.py
index 2e262f0b..9db56398 100644
--- a/test/datagateway_api/db/endpoints/test_create_db.py
+++ b/test/datagateway_api/db/endpoints/test_create_db.py
@@ -20,6 +20,8 @@ def test_valid_create_data(self, flask_test_app_db, valid_db_credentials_header)
                 "doi": "DataGateway API DB Test DOI",
                 "facilityID": 1,
                 "typeID": 1,
+                "fileCount": 1,
+                "fileSize": 6,
             }
             for i in range(2)
         ]
@@ -63,6 +65,8 @@ def test_valid_boundary_create_data(
             "doi": "DataGateway API Test DOI",
             "facilityID": 1,
             "typeID": 1,
+            "fileCount": 3,
+            "fileSize": 2,
         }
 
         test_response = flask_test_app_db.post(
diff --git a/test/datagateway_api/db/endpoints/test_update_by_id_db.py b/test/datagateway_api/db/endpoints/test_update_by_id_db.py
index 20124005..0f5f9f0b 100644
--- a/test/datagateway_api/db/endpoints/test_update_by_id_db.py
+++ b/test/datagateway_api/db/endpoints/test_update_by_id_db.py
@@ -55,4 +55,4 @@ def test_invalid_update_with_id(
             json=invalid_update_json,
         )
 
-        assert test_response.status_code == 400
+        assert test_response.json["doi"] == "_" * 255