diff --git a/fda-database-service/rest-service/src/test/java/at/tuwien/BaseUnitTest.java b/fda-database-service/rest-service/src/test/java/at/tuwien/BaseUnitTest.java
index 55f9f6272b106d475ed0a455104519d0bf4df64e..92d2b08f37acf8a5cde7a5e674d2fd3b0914f0db 100644
--- a/fda-database-service/rest-service/src/test/java/at/tuwien/BaseUnitTest.java
+++ b/fda-database-service/rest-service/src/test/java/at/tuwien/BaseUnitTest.java
@@ -165,6 +165,7 @@ public abstract class BaseUnitTest {
             .internalName(CONTAINER_1_INTERNALNAME)
             .created(CONTAINER_1_CREATED)
             .lastModified(CONTAINER_1_UPDATED)
+            .imageId(IMAGE_1_ID)
             .image(IMAGE_1)
             .build();
 
@@ -175,6 +176,7 @@ public abstract class BaseUnitTest {
             .internalName(CONTAINER_2_INTERNALNAME)
             .created(CONTAINER_2_CREATED)
             .lastModified(CONTAINER_2_UPDATED)
+            .imageId(IMAGE_1_ID)
             .image(IMAGE_1)
             .build();
 
diff --git a/fda-database-service/rest-service/src/test/java/at/tuwien/service/DatabaseServiceIntegrationTest.java b/fda-database-service/rest-service/src/test/java/at/tuwien/service/DatabaseServiceIntegrationTest.java
index e52d86c4db5beac6907ae014395b9b14197abf3c..3d65f8a272f9375d20076330be5d337d07051fcf 100644
--- a/fda-database-service/rest-service/src/test/java/at/tuwien/service/DatabaseServiceIntegrationTest.java
+++ b/fda-database-service/rest-service/src/test/java/at/tuwien/service/DatabaseServiceIntegrationTest.java
@@ -138,6 +138,7 @@ public class DatabaseServiceIntegrationTest extends BaseUnitTest {
         startContainer(CONTAINER_2);
         /* metadata db */
         licenseRepository.save(LICENSE_1);
+        imageRepository.save(IMAGE_1);
         containerRepository.save(CONTAINER_1);
         containerRepository.save(CONTAINER_2);
         USER_1.setPassword(passwordEncoder.encode(USER_1_PASSWORD));