diff --git a/source/java/org/alfresco/repo/workflow/activiti/ActivitiWorkflowEngine.java b/source/java/org/alfresco/repo/workflow/activiti/ActivitiWorkflowEngine.java
index be3312bf7f..5f148c4257 100644
--- a/source/java/org/alfresco/repo/workflow/activiti/ActivitiWorkflowEngine.java
+++ b/source/java/org/alfresco/repo/workflow/activiti/ActivitiWorkflowEngine.java
@@ -51,7 +51,6 @@ import org.activiti.engine.impl.RepositoryServiceImpl;
 import org.activiti.engine.impl.bpmn.behavior.ReceiveTaskActivityBehavior;
 import org.activiti.engine.impl.bpmn.behavior.UserTaskActivityBehavior;
 import org.activiti.engine.impl.bpmn.deployer.BpmnDeployer;
-import org.activiti.engine.impl.bpmn.diagram.ProcessDiagramGenerator;
 import org.activiti.engine.impl.context.Context;
 import org.activiti.engine.impl.form.DefaultTaskFormHandler;
 import org.activiti.engine.impl.form.TaskFormHandler;
@@ -71,6 +70,8 @@ import org.activiti.engine.runtime.Job;
 import org.activiti.engine.runtime.ProcessInstance;
 import org.activiti.engine.task.Task;
 import org.activiti.engine.task.TaskQuery;
+import org.activiti.image.ProcessDiagramGenerator;
+import org.activiti.image.impl.DefaultProcessDiagramGenerator;
 import org.alfresco.model.ContentModel;
 import org.alfresco.repo.i18n.MessageService;
 import org.alfresco.repo.model.Repository;
@@ -1109,11 +1110,11 @@ public class ActivitiWorkflowEngine extends BPMEngine implements WorkflowEngine
             BpmnModel model = repoService.getBpmnModel(pi.getProcessDefinitionId());
 
             if (model != null && model.getLocationMap().size() > 0) 
-            { 
-                return ProcessDiagramGenerator
-                        .generateDiagram(model,
-                                    ActivitiConstants.PROCESS_INSTANCE_IMAGE_FORMAT,
-                                    runtimeService.getActiveActivityIds(processInstanceId)); 
+            {              
+                ProcessDiagramGenerator generator = new DefaultProcessDiagramGenerator();
+                return generator.generateDiagram(model,
+                      ActivitiConstants.PROCESS_INSTANCE_IMAGE_FORMAT,
+                      runtimeService.getActiveActivityIds(processInstanceId)); 
             }
         }
         return null;
diff --git a/source/java/org/alfresco/repo/workflow/activiti/variable/CustomStringVariableType.java b/source/java/org/alfresco/repo/workflow/activiti/variable/CustomStringVariableType.java
index 4b25155992..2086b94a93 100644
--- a/source/java/org/alfresco/repo/workflow/activiti/variable/CustomStringVariableType.java
+++ b/source/java/org/alfresco/repo/workflow/activiti/variable/CustomStringVariableType.java
@@ -32,6 +32,17 @@ import org.activiti.engine.impl.variable.ValueFields;
  */
 public class CustomStringVariableType extends StringType {
     protected static final int MAX_TEXT_LENGTH = 4000;
+    
+    public CustomStringVariableType()
+    {
+        super(MAX_TEXT_LENGTH);
+        
+    }
+    
+    public CustomStringVariableType(int length)
+    {
+        super(length);
+    }
 
     @Override
     public void setValue(Object value, ValueFields valueFields) 
diff --git a/source/test-java/org/alfresco/repo/workflow/activiti/ActivitiWorkflowComponentTest.java b/source/test-java/org/alfresco/repo/workflow/activiti/ActivitiWorkflowComponentTest.java
index 0e927f0e66..7156a9ba5a 100644
--- a/source/test-java/org/alfresco/repo/workflow/activiti/ActivitiWorkflowComponentTest.java
+++ b/source/test-java/org/alfresco/repo/workflow/activiti/ActivitiWorkflowComponentTest.java
@@ -37,7 +37,6 @@ import java.util.List;
 import java.util.Map;
 
 import org.activiti.engine.history.HistoricProcessInstance;
-import org.activiti.engine.impl.util.ClockUtil;
 import org.activiti.engine.repository.ProcessDefinition;
 import org.activiti.engine.runtime.Execution;
 import org.activiti.engine.runtime.Job;
@@ -585,7 +584,6 @@ public class ActivitiWorkflowComponentTest extends AbstractActivitiComponentTest
         WorkflowDefinition def = deployTestAdhocDefinition();
         
         Date startTime = new SimpleDateFormat("dd-MM-yyy hh:mm:ss").parse("01-01-2011 12:11:10");
-        ClockUtil.setCurrentTime(startTime);
         
         // Add some variables which should be used in the WorkflowInstance
         Map<String, Object> variables = new HashMap<String, Object>();
@@ -621,8 +619,6 @@ public class ActivitiWorkflowComponentTest extends AbstractActivitiComponentTest
         
         assertEquals(startTime, workflowInstance.getStartDate());
         
-        // Reset current time used in activiti
-        ClockUtil.setCurrentTime(null);
     }
     
     @Test
@@ -631,7 +627,6 @@ public class ActivitiWorkflowComponentTest extends AbstractActivitiComponentTest
         WorkflowDefinition def = deployTestAdhocDefinition();
         
         Date startTime = new SimpleDateFormat("dd-MM-yyy hh:mm:ss").parse("01-01-2011 01:02:03");
-        ClockUtil.setCurrentTime(startTime);
         
         // Add some variables which should be used in the WorkflowInstance
         Map<String, Object> variables = new HashMap<String, Object>();
@@ -649,7 +644,6 @@ public class ActivitiWorkflowComponentTest extends AbstractActivitiComponentTest
             ActivitiConstants.ENGINE_ID, processInstance.getProcessInstanceId());
         
         Date endTime = new SimpleDateFormat("dd-MM-yyy hh:mm:ss").parse("01-01-2011 02:03:04");
-        ClockUtil.setCurrentTime(endTime);
         
         // Finish the task
         Task task = taskService.createTaskQuery().processInstanceId(processInstance.getId()).singleResult();
@@ -673,8 +667,6 @@ public class ActivitiWorkflowComponentTest extends AbstractActivitiComponentTest
         assertEquals(3, workflowInstance.getPriority().intValue());
         assertEquals(startTime, workflowInstance.getStartDate());
         
-        // Reset current time used in activiti
-        ClockUtil.setCurrentTime(null);
     }
     
     @Test