diff --git a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/testcase-registration-context.py b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/testcase-registration-context.py
index 0665af678108548c5b9952089f8315f5d9b9e5c7..e80f0f6049dfbc0bd30b0937d04acb6dfc1ae5c6 100644
--- a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/testcase-registration-context.py
+++ b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/testcase-registration-context.py
@@ -4,7 +4,7 @@ jythonHookTestTool = JythonHookTestTool.createInTest()
 
 execfile("sourceTest/java/ch/systemsx/cisd/etlserver/registrator/simple-transaction.py")
 
-transaction.getRegistrationContext().put("body","1")
+transaction.getRegistrationContext().getPersistentMap().put("body","1")
 
 def rollback_transaction(service, transaction, algorithmRunner, throwable):
     jythonHookTestTool.log("rollback_transaction")
@@ -15,7 +15,7 @@ def pre_metadata_registration(context):
     assert_context_content(context, "pre_metadata_registration", "pre_metadata_registration", None);
     assert_context_content(context, "pre_metadata_registration", "post_storage", None);
 
-    context.put("pre_metadata_registration", "2")
+    context.getPersistentMap().put("pre_metadata_registration", "2")
     jythonHookTestTool.log("pre_metadata_registration")
 
 def post_metadata_registration(context):
@@ -24,7 +24,7 @@ def post_metadata_registration(context):
     assert_context_content(context, "post_metadata_registration", "pre_metadata_registration", "2");
     assert_context_content(context, "post_metadata_registration", "post_storage", None);
 
-    context.put("post_metadata_registration", "3")
+    context.getPersistentMap().put("post_metadata_registration", "3")
     jythonHookTestTool.log("post_metadata_registration")
 
 def post_storage(context):
@@ -33,11 +33,11 @@ def post_storage(context):
     assert_context_content(context, "post_storage", "pre_metadata_registration", "2");
     assert_context_content(context, "post_storage", "post_metadata_registration", "3");
 
-    context.put("post_storage", "4")
+    context.getPersistentMap().put("post_storage", "4")
     jythonHookTestTool.log("post_storage")
     
 def assert_context_content(context, caller, name, expected):
-    value = context.get(name)
+    value = context.getPersistentMap().get(name)
     if (value != expected):
         if (value != None and expected != None and type(value) != type(expected)):
             value = "%s:%s" % (type(value), value)
diff --git a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/v2-retry-process.py b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/v2-retry-process.py
index 4f1c72c98a4f1f69f52eeecd55fe47048dab70a2..27cb34b0a11c5d63d35c1ac7fa7ff9ffee0cb784 100644
--- a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/v2-retry-process.py
+++ b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/v2-retry-process.py
@@ -1,16 +1,16 @@
 execfile("sourceTest/java/ch/systemsx/cisd/etlserver/registrator/all-hooks.py")
 
-def should_retry_processing(map, error):
-  lastResult = map.get("RETRY_COUNT")
+def should_retry_processing(context, error):
+  lastResult = context.getPersistentMap().get("RETRY_COUNT")
   if (lastResult == None):
     lastResult = 0
-  map.put("RETRY_COUNT", lastResult + 1)
+  context.getPersistentMap().put("RETRY_COUNT", lastResult + 1)
   print "RETRY TRANSACTION PUTS THE SCORE %d" % (lastResult + 1)
   return True
 
 def process(transaction):
   incoming = transaction.getIncoming()
-  key = transaction.getRegistrationContext().get("RETRY_COUNT");
+  key = transaction.getRegistrationContext().getPersistentMap().get("RETRY_COUNT");
   if (key == None):
     key = 1
   dataSet = transaction.createNewDataSet()
diff --git a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/v2-testcase-registration-context.py b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/v2-testcase-registration-context.py
index 9f5d0917ce80132d28510a464fa5446a79ed47b6..e81c6db67dcc04d1ec83adec3405decc1839ddab 100644
--- a/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/v2-testcase-registration-context.py
+++ b/datastore_server/sourceTest/java/ch/systemsx/cisd/etlserver/registrator/v2-testcase-registration-context.py
@@ -7,7 +7,7 @@ def process(transaction):
   transaction.moveFile(transaction.getIncoming().getPath() + '/sub_data_set_1', dataSet)
   dataSet.setDataSetType('O1')
   dataSet.setExperiment(transaction.getExperiment('/SPACE/PROJECT/EXP'))
-  transaction.getRegistrationContext().put("body","1")
+  transaction.getRegistrationContext().getPersistentMap().put("body","1")
 
 def rollback_transaction(service, transaction, algorithmRunner, throwable):
     jythonHookTestTool.log("rollback_transaction")
@@ -18,7 +18,7 @@ def pre_metadata_registration(context):
     assert_context_content(context, "pre_metadata_registration", "pre_metadata_registration", None);
     assert_context_content(context, "pre_metadata_registration", "post_storage", None);
 
-    context.put("pre_metadata_registration", "2")
+    context.getPersistentMap().put("pre_metadata_registration", "2")
     jythonHookTestTool.log("pre_metadata_registration")
 
 def post_metadata_registration(context):
@@ -27,7 +27,7 @@ def post_metadata_registration(context):
     assert_context_content(context, "post_metadata_registration", "pre_metadata_registration", "2");
     assert_context_content(context, "post_metadata_registration", "post_storage", None);
 
-    context.put("post_metadata_registration", "3")
+    context.getPersistentMap().put("post_metadata_registration", "3")
     jythonHookTestTool.log("post_metadata_registration")
 
 def post_storage(context):
@@ -36,11 +36,11 @@ def post_storage(context):
     assert_context_content(context, "post_storage", "pre_metadata_registration", "2");
     assert_context_content(context, "post_storage", "post_metadata_registration", "3");
 
-    context.put("post_storage", "4")
+    context.getPersistentMap().put("post_storage", "4")
     jythonHookTestTool.log("post_storage")
     
 def assert_context_content(context, caller, name, expected):
-    value = context.get(name)
+    value = context.getPersistentMap().get(name)
     if (value != expected):
         if (value != None and expected != None and type(value) != type(expected)):
             value = "%s:%s" % (type(value), value)