1 files changed, 4 insertions, 0 deletions
diff --git a/Project/ControlElementContainer.cpp b/Project/ControlElementContainer.cpp
index 3c420e7..042f7f2 100644
--- a/Project/ControlElementContainer.cpp
+++ b/Project/ControlElementContainer.cpp
@@ -46,6 +46,8 @@ void ControlElementContainer::FillContainer(ControlEditor* editor)
             m_sumList.push_back(sum);
         } else if(TransferFunction* tf = dynamic_cast<TransferFunction*>(*it)) {
             m_tfList.push_back(tf);
+        } else if(Divider* divider = dynamic_cast<Divider*>(*it)) {
+            m_dividerList.push_back(divider);
         }
     }
 }
@@ -90,6 +92,8 @@ void ControlElementContainer::FillContainer(std::vector<ControlElement*> control
             m_sumList.push_back(sum);
         } else if(TransferFunction* tf = dynamic_cast<TransferFunction*>(*it)) {
             m_tfList.push_back(tf);
+        } else if(Divider* divider = dynamic_cast<Divider*>(*it)) {
+            m_dividerList.push_back(divider);
         }
     }
 }