diff --git a/AIStudio.Wpf.DiagramApp/ViewModels/PageViewModel.cs b/AIStudio.Wpf.DiagramApp/ViewModels/PageViewModel.cs
index c26fab7..e3168d3 100644
--- a/AIStudio.Wpf.DiagramApp/ViewModels/PageViewModel.cs
+++ b/AIStudio.Wpf.DiagramApp/ViewModels/PageViewModel.cs
@@ -430,7 +430,7 @@ namespace AIStudio.Wpf.DiagramApp.ViewModels
 
             foreach (var viewModel in DiagramViewModels)
             {
-                DiagramItem diagramItem = new DiagramItem(viewModel);        
+                DiagramItem diagramItem = new DiagramItem(viewModel, ext);        
                 diagramDocument.DiagramItems.Add(diagramItem);
             }