diff --git a/include/model/model.hpp b/include/model/model.hpp
index 6f234e3..9535b79 100644
--- a/include/model/model.hpp
+++ b/include/model/model.hpp
@@ -39,7 +39,7 @@ class Model {
 
     bool validate();
 
-  private:
+  protected:
     Layers::Input*  inputLayer;
     Layers::Output* outputLayer;
 
diff --git a/include/model/module.hpp b/include/model/module.hpp
index c3992b1..6fd5ada 100644
--- a/include/model/module.hpp
+++ b/include/model/module.hpp
@@ -23,7 +23,7 @@ class Module : public Layers::SequentialLayer {
 
     const std::unordered_map<std::string, Layers::SequentialLayer*>& getLayers() const;
 
-  private:
+  protected:
     int inputSize;
     int inputChannels;