diff --git a/Program.cs b/Program.cs
index 3f1ff07..59eb0c6 100644
--- a/Program.cs
+++ b/Program.cs
@@ -14,8 +14,8 @@ if (string.IsNullOrEmpty(workflowViewFile) || !Directory.Exists(projectDirectory
 
 var workflow = JsonSerializer.Deserialize<S8nWorkflow>(File.OpenRead(workflowViewFile), JsonSerializerOptions.Web)!;
 
-var edgesFile = Path.Combine("/sources", $"{workflow.Name}.Edges.cs");
-var nodesFile = Path.Combine("/sources", $"{workflow.Name}.Nodes.cs");
+var edgesFile = Path.Combine(projectDirectory, $"{workflow.Name}.Edges.cs");
+var nodesFile = Path.Combine(projectDirectory, $"{workflow.Name}.Nodes.cs");
 
 // Linking
 const string templateEdges = """