diff --git a/MVMCoreUI/Behaviors/AddRemoveMoleculeBehavior.swift b/MVMCoreUI/Behaviors/AddRemoveMoleculeBehavior.swift
index ee4d9587..a3be96f5 100644
--- a/MVMCoreUI/Behaviors/AddRemoveMoleculeBehavior.swift
+++ b/MVMCoreUI/Behaviors/AddRemoveMoleculeBehavior.swift
@@ -75,7 +75,7 @@ public class AddRemoveMoleculesBehavior: PageCustomActionHandlerBehavior, PageMo
         self.delegate = delegateObject
     }
     
-    public func onPageNew(rootMolecules: [MoleculeModelProtocol], _ delegateObject: MVMCoreUIDelegateObject) {
+    public func onPageNew(rootMolecules: [MoleculeModelProtocol], _ delegateObject: MVMCoreUIDelegateObject?) {
         guard let list = delegate?.moleculeListDelegate else { return }
         for case let model as (MoleculeModelProtocol & ListItemModelProtocol & AddMolecules) in rootMolecules {
             if let moleculesToAdd = model.getRecursiveMoleculesToAdd(),
diff --git a/MVMCoreUI/Behaviors/Protocols/PageBehaviorProtocol.swift b/MVMCoreUI/Behaviors/Protocols/PageBehaviorProtocol.swift
index 76bc7f84..eadd5e4e 100644
--- a/MVMCoreUI/Behaviors/Protocols/PageBehaviorProtocol.swift
+++ b/MVMCoreUI/Behaviors/Protocols/PageBehaviorProtocol.swift
@@ -19,7 +19,7 @@ public protocol PageBehaviorProtocol: ModelHandlerProtocol {
 
 public protocol PageMoleculeTransformationBehavior: PageBehaviorProtocol {
     
-    func onPageNew(rootMolecules: [MoleculeModelProtocol], _ delegateObject: MVMCoreUIDelegateObject)
+    func onPageNew(rootMolecules: [MoleculeModelProtocol], _ delegateObject: MVMCoreUIDelegateObject?)
 }
 
 public protocol PageVisibilityBehavior: PageBehaviorProtocol {