diff --git a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/JythonTopLevelDataSetRegistratorTest.java b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/JythonTopLevelDataSetRegistratorTest.java
index c686923c74ab2e5161cba7c5ffbf5721272e4358..78f2e1784a1c524a273ddf3cd6222bd399162d49 100644
--- a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/JythonTopLevelDataSetRegistratorTest.java
+++ b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/JythonTopLevelDataSetRegistratorTest.java
@@ -533,7 +533,6 @@ public class JythonTopLevelDataSetRegistratorTest extends AbstractFileSystemTest
         context.assertIsSatisfied();
     }
 
-    // TODO KE: make me work !
     @Test
     public void testTransactionWithDataSetUpdate()
     {
@@ -600,6 +599,8 @@ public class JythonTopLevelDataSetRegistratorTest extends AbstractFileSystemTest
                 new PropertyBuilder("newProp").value("newValue").getProperty();
         assertEquals(Arrays.asList(propertyChanged).toString(), dataSetUpdate.getProperties()
                 .toString());
+        assertEquals(EXPERIMENT_IDENTIFIER, dataSetUpdate.getExperimentIdentifierOrNull()
+                .toString());
 
         File datasetLocation =
                 DatasetLocationUtil.getDatasetLocationPath(workingDirectory, DATA_SET_CODE,
diff --git a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/transaction-with-dataset-update.py b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/transaction-with-dataset-update.py
index 08212defd1bd6948ac2f982712aae42e3d8424c0..c7fd2f0cafa99f142bce2cc20c3909049bccde06 100644
--- a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/transaction-with-dataset-update.py
+++ b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/transaction-with-dataset-update.py
@@ -12,4 +12,5 @@ newContents = list(container.containedDataSetCodes)
 newContents.append(dataSet.dataSetCode)
 container.setContainedDataSetCodes(newContents)
 container.setPropertyValue("newProp", "newValue")
+container.setExperiment(experiment)