diff --git a/sanofi/sourceTest/java/ch/systemsx/cisd/sanofi/dss/test/SanofiDropboxJythonTest.java b/sanofi/sourceTest/java/ch/systemsx/cisd/sanofi/dss/test/SanofiDropboxJythonTest.java
index f97a702d058a33d931cdd693ad6cf8eb9b8aa483..69ac16ce7fe727bda1e82765a7dd00c415711bb8 100644
--- a/sanofi/sourceTest/java/ch/systemsx/cisd/sanofi/dss/test/SanofiDropboxJythonTest.java
+++ b/sanofi/sourceTest/java/ch/systemsx/cisd/sanofi/dss/test/SanofiDropboxJythonTest.java
@@ -847,6 +847,8 @@ public class SanofiDropboxJythonTest extends AbstractJythonDataSetHandlerTest
                                     ANALYSIS_DATA_SET_FILE_NAME));
 
                     allowing(openBisService).setStorageConfirmed(with(any(String.class)));
+                    allowing(openBisService).markDataSetForRegistration(with(any(String.class)));
+                    
                 }
             });
     }