diff --git a/src/core/Elsa.Abstractions/Services/CompositeActivity.cs b/src/core/Elsa.Abstractions/Services/CompositeActivity.cs
index d7733fe96..23aed2b29 100644
--- a/src/core/Elsa.Abstractions/Services/CompositeActivity.cs
+++ b/src/core/Elsa.Abstractions/Services/CompositeActivity.cs
@@ -33,7 +33,7 @@ namespace Elsa.Services
             IsScheduled = false;
             await OnExitAsync(context);
 
-            var finishOutput = context.GetInput<FinishOutput>();
+            var finishOutput = context.Input as FinishOutput;
             var outcomes = new List<string> { OutcomeNames.Done };
             var output = default(object?);