diff --git a/NeuralNetwork/Layer/Interfaces/Layer.mo b/NeuralNetwork/Layer/Interfaces/Layer.mo
index 23b8060..e562111 100644
--- a/NeuralNetwork/Layer/Interfaces/Layer.mo
+++ b/NeuralNetwork/Layer/Interfaces/Layer.mo
@@ -1,6 +1,6 @@
 within NeuralNetwork.Layer.Interfaces;
 
-partial model Layer
+partial block Layer
   // Topology of the neural network
   parameter Integer numInputs   "Specification of the inputs of the layer";
   parameter Integer numNeurons  "Number of neurons in the layer";