diff --git a/src/modules/Elsa.JavaScript/TypeDefinitions/Providers/ActivityOutputFunctionsDefinitionProvider.cs b/src/modules/Elsa.JavaScript/TypeDefinitions/Providers/ActivityOutputFunctionsDefinitionProvider.cs
index bdb3e0e9e..a0d23fd1c 100644
--- a/src/modules/Elsa.JavaScript/TypeDefinitions/Providers/ActivityOutputFunctionsDefinitionProvider.cs
+++ b/src/modules/Elsa.JavaScript/TypeDefinitions/Providers/ActivityOutputFunctionsDefinitionProvider.cs
@@ -14,11 +14,13 @@ internal class ActivityOutputFunctionsDefinitionProvider : FunctionDefinitionPro
 {
     private readonly IActivityVisitor _activityVisitor;
     private readonly IActivityRegistry _activityRegistry;
+    private readonly IIdentityGraphService _identityGraphService;
 
-    public ActivityOutputFunctionsDefinitionProvider(IActivityVisitor activityVisitor, IActivityRegistry activityRegistry)
+    public ActivityOutputFunctionsDefinitionProvider(IActivityVisitor activityVisitor, IActivityRegistry activityRegistry, IIdentityGraphService identityGraphService)
     {
         _activityVisitor = activityVisitor;
         _activityRegistry = activityRegistry;
+        _identityGraphService = identityGraphService;
     }
 
     protected override async ValueTask<IEnumerable<FunctionDefinition>> GetFunctionDefinitionsAsync(TypeDefinitionContext context)
@@ -27,6 +29,8 @@ internal class ActivityOutputFunctionsDefinitionProvider : FunctionDefinitionPro
         var nodes = (await _activityVisitor.VisitAsync(context.Workflow.Root, context.CancellationToken)).Flatten().Distinct().ToList();
         var activitiesWithOutputs = nodes.GetActivitiesWithOutputs(_activityRegistry);
         var definitions = new List<FunctionDefinition>();
+        
+        _identityGraphService.AssignIdentities(nodes);
 
         foreach (var (activity, activityDescriptor) in activitiesWithOutputs)
         {