diff --git a/agui2/src/Graphics/GraphicsBox.cpp b/agui2/src/Graphics/GraphicsBox.cpp
index 4ebc170fa926bfe8e021a1c6449bc00ecb319533..75895c91625ccfd2c50c5bdbf2ab17b8bb1bdb66 100644
--- a/agui2/src/Graphics/GraphicsBox.cpp
+++ b/agui2/src/Graphics/GraphicsBox.cpp
@@ -49,3 +49,11 @@ void GraphicsBox::paint(QPainter* painter, const QStyleOptionGraphicsItem* optio
     painter->fillRect(this->boundRect, this->color);
     painter->drawText(this->boundRect, Qt::AlignCenter, this->text);
 }
+
+const std::vector<InputConnectionBox*>& GraphicsBox::getInputConnectionBoxes() const {
+    return inputConnectionBoxes;
+}
+
+OutputConnectionBox* GraphicsBox::getOutputConnectionBox() const {
+    return outputConnectionBox;
+}
\ No newline at end of file
diff --git a/agui2/src/Graphics/GraphicsBox.hpp b/agui2/src/Graphics/GraphicsBox.hpp
index 8baa6e23efa7142369269ddacc773d921e7824ec..9e761320be4a7d3459876eb07a674c05eda1240f 100644
--- a/agui2/src/Graphics/GraphicsBox.hpp
+++ b/agui2/src/Graphics/GraphicsBox.hpp
@@ -6,6 +6,8 @@
 #include <Models/ModelBox.hpp>
 
 class ConnectionBox;
+class InputConnectionBox;
+class OutputConnectionBox;
 
 class GraphicsBox : public QGraphicsObject {
     Q_OBJECT
@@ -17,6 +19,9 @@ public:
 
     ModelBox* getModelBox() const { return this->modelBox.get(); }
 
+    const std::vector<InputConnectionBox*>& getInputConnectionBoxes() const;
+    OutputConnectionBox* getOutputConnectionBox() const;
+
 protected:
     QColor color;
 
@@ -27,8 +32,8 @@ private:
 
     std::unique_ptr<ModelBox> modelBox;
 
-    std::vector<ConnectionBox*> inputConnectionBoxes;
-    ConnectionBox* outputConnectionBox;
+    std::vector<InputConnectionBox*> inputConnectionBoxes;
+    OutputConnectionBox* outputConnectionBox;
 };