diff --git a/src/Neuron/NeuronLogistic.cpp b/src/Neuron/NeuronLogistic.cpp
index 10e586fc30be5706257025e8f48fc8bfbbd0c74e..b3b398dc94c3b9c576eec98d2e7e9f8665940f3c 100644
--- a/src/Neuron/NeuronLogistic.cpp
+++ b/src/Neuron/NeuronLogistic.cpp
@@ -7,6 +7,7 @@
 
 NeuronLogistic::NeuronLogistic(double a, double b) {
 
+    this->n_activation_function_parameters = 2;
     this->activation_function_parameters = new double[2];
 
     this->activation_function_parameters[0] = a;
diff --git a/src/Neuron/NeuronTanh.cpp b/src/Neuron/NeuronTanh.cpp
index b6cd25a4353261d96c9c232d4a228e579c8179cf..74d9258329f11393efa367acf2963603169d8f1d 100644
--- a/src/Neuron/NeuronTanh.cpp
+++ b/src/Neuron/NeuronTanh.cpp
@@ -6,6 +6,7 @@
 
 NeuronTanh::NeuronTanh(double a) {
 
+    this->n_activation_function_parameters = 1;
     this->activation_function_parameters = new double[1];
     this->activation_function_parameters[0] = a;