diff --git a/integration-tests/templates/test_openbis_sync/core-plugins/openbis1/1/dss/drop-boxes/entity-reg/dropbox.py b/integration-tests/templates/test_openbis_sync/core-plugins/openbis1/1/dss/drop-boxes/entity-reg/dropbox.py
index 42ddb292c447003df44b9b929f100c7b2eb9f97c..40ffc6e2d65caf072711bbc061a818c5e1fd1998 100644
--- a/integration-tests/templates/test_openbis_sync/core-plugins/openbis1/1/dss/drop-boxes/entity-reg/dropbox.py
+++ b/integration-tests/templates/test_openbis_sync/core-plugins/openbis1/1/dss/drop-boxes/entity-reg/dropbox.py
@@ -43,8 +43,7 @@ def createDsWithPermId(transaction, sample, code, ds_type):
     return ds
 
 def registerContainerDS(transaction, experiment, comp_ds_code):
-    container = transaction.createNewDataSet("HCS_IMAGE_CONTAINER_RAW", "CONTAINER_DS1")
-    container.setDataSetKind(DataSetKind.CONTAINER)
+    container = transaction.createNewDataSet("HCS_IMAGE_CONTAINER_RAW", "CONTAINER_DS1", DataSetKind.CONTAINER)
     container.setExperiment(experiment)
     container.setContainedDataSetCodes([comp_ds_code])