diff --git a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/TransferredDataSetHandlerTest.java b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/TransferredDataSetHandlerTest.java
index 961f3be0578db2bb0c3697e65172de50a7b65b41..f607f96ca81e84ff1efe528aa4c0e5eb2b2c1c4e 100644
--- a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/TransferredDataSetHandlerTest.java
+++ b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/TransferredDataSetHandlerTest.java
@@ -558,6 +558,8 @@ public final class TransferredDataSetHandlerTest extends AbstractFileSystemTestC
                     one(transaction).commit();
 
                     one(shareIdManager).setShareId(DATA_SET_CODE, "1");
+                    
+                    allowing(limsService).setStorageConfirmed(with(equal(SESSION_TOKEN)), with(equal(DATA_SET_CODE)));
                 }
             });
         final LogMonitoringAppender appender =
@@ -606,6 +608,8 @@ public final class TransferredDataSetHandlerTest extends AbstractFileSystemTestC
                     one(transaction).commit();
 
                     one(shareIdManager).setShareId(DATA_SET_CODE, "1");
+                    
+                    allowing(limsService).setStorageConfirmed(with(equal(SESSION_TOKEN)), with(equal(DATA_SET_CODE)));
                 }
             });
         final LogMonitoringAppender appender =
@@ -847,6 +851,8 @@ public final class TransferredDataSetHandlerTest extends AbstractFileSystemTestC
                     one(transaction).commit();
 
                     one(shareIdManager).setShareId(DATA_SET_CODE, "1");
+
+                    allowing(limsService).setStorageConfirmed(with(equal(SESSION_TOKEN)), with(equal(DATA_SET_CODE)));
                 }
             });
         final LogMonitoringAppender appender =