diff --git a/eu_basynthec/sourceTest/java/eu/basynthec/cisd/dss/AbstractBaSynthecDataSetRegistratorTest.java b/eu_basynthec/sourceTest/java/eu/basynthec/cisd/dss/AbstractBaSynthecDataSetRegistratorTest.java
index 3091229a84a9398716d32a9a06aa07e634eb495a..f374f8194b1321ce594d200064d915e972a4a8c6 100644
--- a/eu_basynthec/sourceTest/java/eu/basynthec/cisd/dss/AbstractBaSynthecDataSetRegistratorTest.java
+++ b/eu_basynthec/sourceTest/java/eu/basynthec/cisd/dss/AbstractBaSynthecDataSetRegistratorTest.java
@@ -121,7 +121,10 @@ public abstract class AbstractBaSynthecDataSetRegistratorTest extends
                             new File(new File(stagingDirectory, tsvDataSetCode), "tsv"));
 
                     one(openBisService).performEntityOperations(with(atomicatOperationDetails));
+
                     will(returnValue(new AtomicEntityOperationResult()));
+
+                    allowing(openBisService).setStorageConfirmed(with(any(String.class)));
                 }
             });
         return atomicatOperationDetails;