diff --git a/src/samples/console/Elsa.Samples.LoopingWorkflows/Workflows/ForWorkflow.cs b/src/samples/console/Elsa.Samples.LoopingWorkflows/Workflows/ForWorkflow.cs
index ab4a71f6c..07d51be90 100644
--- a/src/samples/console/Elsa.Samples.LoopingWorkflows/Workflows/ForWorkflow.cs
+++ b/src/samples/console/Elsa.Samples.LoopingWorkflows/Workflows/ForWorkflow.cs
@@ -22,7 +22,7 @@ public static class ForWorkflow
                 new WriteLine("Counting down from 10 to 1:"),
                 new For(10, 1)
                 {
-                    CurrentValue = new Output<MemoryBlockReference?>(currentValueVariable),
+                    CurrentValue = new Output<object?>(currentValueVariable),
                     Body = new Sequence
                     {
                         Activities =
diff --git a/test/integration/Elsa.IntegrationTests/Activities/Break/BreakForWorkflow.cs b/test/integration/Elsa.IntegrationTests/Activities/Break/BreakForWorkflow.cs
index ddac39090..5108e75f9 100644
--- a/test/integration/Elsa.IntegrationTests/Activities/Break/BreakForWorkflow.cs
+++ b/test/integration/Elsa.IntegrationTests/Activities/Break/BreakForWorkflow.cs
@@ -1,9 +1,7 @@
-using Elsa.Expressions.Models;
 using Elsa.Workflows.Core.Abstractions;
 using Elsa.Workflows.Core.Activities;
 using Elsa.Workflows.Core.Contracts;
 using Elsa.Workflows.Core.Models;
-using Elsa.Workflows.Core.Services;
 
 namespace Elsa.IntegrationTests.Activities;
 
@@ -20,7 +18,7 @@ class BreakForWorkflow : WorkflowBase
                 new WriteLine("Start"),
                 new For(0, 3)
                 {
-                    CurrentValue = new Output<MemoryBlockReference?>(currentValue),
+                    CurrentValue = new Output<object?>(currentValue),
                     Body = new Sequence
                     {
                         Activities =